U-net-pytorch-segmentation

U-net,pytorch,Image semantic segmentation

数据准备

项目文件分布如下

  --project
  	main.py
  	 --data
   		--train
   		--val


## 模型训练

python main.py train


## 测试模型训练
加载权重，默认保存最后一个权重

python main.py test

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
.idea		.idea
__pycache__		__pycache__
data		data
README.md		README.md
dataset.py		dataset.py
loss.npy		loss.npy
main.py		main.py
note		note
unet.py		unet.py