Unet-pytorch

a unet implementation for Image semantic segmentation

参考网上的Unet做分割的代码,做了一个针对kaggle地盐识别的,请去以下地址获取数据集:

https://www.kaggle.com/c/tgs-salt-identification-challenge

Unet参考以下Github项目:

https://github.com/Qiuyan918/Unet_Implementation_PyTorch

感谢Qiuyan918的分享!


项目结构如下:
——-salt_unet——–
|

|data(place your dataset here)

|

|

|

|

|

|config.py

|

|model.py

|

|train.py

|

|dataset.py

GitHub

View Github