├── LICENSE ├── README.md ├── model ├── s2block.py └── u2net.py ├── requirements.txt ├── show_image ├── cave.png ├── cave_visual.png ├── s2block.png ├── u2net.png ├── wv2.png ├── wv3.png └── wv3_visual.png ├── test.py ├── train.py ├── utils ├── load_test_data.py └── load_train_data.py └── weights └── 200000_iter.pth /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/README.md -------------------------------------------------------------------------------- /model/s2block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/model/s2block.py -------------------------------------------------------------------------------- /model/u2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/model/u2net.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/requirements.txt -------------------------------------------------------------------------------- /show_image/cave.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/cave.png -------------------------------------------------------------------------------- /show_image/cave_visual.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/cave_visual.png -------------------------------------------------------------------------------- /show_image/s2block.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/s2block.png -------------------------------------------------------------------------------- /show_image/u2net.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/u2net.png -------------------------------------------------------------------------------- /show_image/wv2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/wv2.png -------------------------------------------------------------------------------- /show_image/wv3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/wv3.png -------------------------------------------------------------------------------- /show_image/wv3_visual.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/show_image/wv3_visual.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/train.py -------------------------------------------------------------------------------- /utils/load_test_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/utils/load_test_data.py -------------------------------------------------------------------------------- /utils/load_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/utils/load_train_data.py -------------------------------------------------------------------------------- /weights/200000_iter.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PSRben/U2Net/HEAD/weights/200000_iter.pth --------------------------------------------------------------------------------