pytorch FCN easiest demo
不断更新中~
这个repo是在读论文时的一个pytorch简单复现,数据集很小,是一些随机背景上的一些包的图片(所有数据集大小一共不到80M),如下图
关于此数据集详细信息,见
根据论文实现了FCN32s、FCN16s、FCN8s和FCNs
部分代码参考了
使用visdom可视化,运行了20个epoch后的可视化如下图:
1.如何运行
1.1 我的运行环境
Windows 10
CUDA 9.x (可选)
Anaconda 3 (numpy、os、datetime、matplotlib)
pytorch == 0.4.1 or 1.0
torchvision == 0.2.1
visdom == 0.1.8.5
OpenCV-Python == 3.4.1
1.2 具体操作
打开终端,输入
python -m visdom.server
打
1