PyTorch Implementation of Fully Convolutional Networks (a very simple and easy demo).
不断更新中~
这个repo是在读论文Fully Convolutional Networks for Semantic Segmentation时的一个pytorch简单复现,数据集很小,是一些随机背景上的一些包的图片(所有数据集大小一共不到80M),如下图
关于此数据集详细信息,见数据集
根据论文实现了FCN32s、FCN16s、FCN8s和FCNs
部分代码参考了这个repo
使用visdom可视化,运行了20个epoch后的可视化如下图:
sh python -m visdom.server
sh python train.py
http://localhost:8097/来使用
visdom可视化