Simple scripts for testing U-2-Net models in Keras.
Based on https://github.com/NathanUA/U-2-Net and https://github.com/shreyas-bk/U-2-Net-Keras
Keras u2net, u2netp and u2net_portrait models were converted from the original pytorch models.

1. Clone this repo to local

git clone https://github.com/Voinic/u2net-keras.git

2. Get models from here

3. Test u2net/u2netp (background substraction) models

python u2net-keras.py images/input.jpg images/output.png

4. Test u2net_portrait model

python u2net_portrait-keras.py images/input_portrait.jpg images/output_portrait.png

Background substraction model


Portrait creation model

• Dont work with tensorflow without CUDA support.