Our great sponsors
-
U-2-Net
The code for our newly accepted paper in Pattern Recognition 2020: "U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection."
-
WorkOS
The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
Since you want to do semantic segmentation with a U-net, I would recommend tearing apart this project: https://github.com/levindabhi/cloth-segmentation
There, the author has the same goal as you do, and has a train.py and instructions. You can reach out to the author and ask questions either in the issues section or perhaps email directly. Many times people are very helpful when you show interest in their work. The neural network it is based on (U2-net) is very easy to get running by the way, and has lots of use cases: https://github.com/xuebinqin/U-2-Net