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