- 通过Pytorch简单实现了Fully Convolutional Networks for Semantic Segmentation
|--dataset
| |--dataset.py:pspnet数据集操作
| |--utils.py:数据处理配置文件
|--nets
| |--vggnet.py:vgg16特征提取主干网络
| |--resnet.py:resnet34特征提取主干网络
| |--fcn.py: FCN网络模型
|--utils
| |--callbacks.py:记录日志相关操作
| |--loss_utils.py:损失函数
| |--metrics_utils.py:评价指标相关
| |--train_utils.py:训练相关配置
|--train.py:训练脚本
|--predict.py:预测脚本
|--get_miou.py:计算模型mIoU
- VOC数据下载地址: http://host.robots.ox.ac.uk/pascal/VOC/
numpy==1.21.2
torch==1.9.1
torchvision==0.11.1
pillow==8.3.1
opencv-python==4.5.4.58
scipy==1.7.2
matplotlib==3.4.3
tqdm==4.62.3
