U-2-Net-Keras

Keras version of U2-Net: Going Deeper with Nested U-Structure for Salient Object Detection

Please star if this was helpful

Link to colab notebook: U-2-Netp Demonstration Colab-PyTorch2Keras.ipynb

Note: both models require GPU support as there are some issues during conversion for CPU only

Download saved model (keras version of u2netp) from Drive

Download saved model (keras version of u2net) from Drive

Inference

Inference code can be found in the notebook itself or minimal testing scripts can be found at https://github.com/Voinic/u2net-keras (includes GPU support for u2net_portrait inference as well)

TODO:

• add rectified inference cell that doesn't use torch
• add CPU supported weights trained on Keras model