shreyas-bk / u2netdemo

Demonstration using Google Colab to show how U-2-NET can be used for Background Removal, Changing Backgrounds, Bounding Box Creation, Salient Feature Highlighting and Salient Object Cropping.
MIT License
198 stars 35 forks source link

An issue on deploying to Tensorflow savedmodel #1

Open rose-jinyang opened 4 years ago

rose-jinyang commented 4 years ago

Hello How are you? Thanks for contributing this project. I tried to deploy this pre-trained model to Tensorflow savedmodel but failed. Could u provide tensorflow savedmodel too? Thanks

shreyas-bk commented 4 years ago

Hi @rose-jinyang , I haven't tried implementing the model in tensorflow, but I will let you know if I do. Maybe contacting the original authors of the U^2-Net model may help.

rose-jinyang commented 4 years ago

I need a Tensorflow savedmodel. Thanks

shreyas-bk commented 4 years ago

I will try

shreyas-bk commented 3 years ago

Please have a look at this repository Implementation of the paper - U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection in TensorFlow.