Shreyas Bhat Kera

Implemented a quick fix #4 for this error: FileNotFoundError: [Errno 2] No such file or directory: '/content/U-2-Net/results/.ipynb_checkpo.png'. Let me know if it works.

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

I will try

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.](https://github.com/Akhilesh64/Image-Segmentation-U-2-Net)

I am not sure this can be done with this model, I don't get good results when I try

It's saying ctx is not defined for me, am I missing something?

The ctx not defined error is solved, but there's a new error. It seems like save_for_backward can only save tensors, not dim(int) and consensus_type(str): `TypeError: save_for_backward can only save variables,...

I've created a PR on your fork with the code that works for me: [PR](https://github.com/tchainzzz/TPN/pull/1)