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 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