文章目录
1. 代码使用
本博文讲解的是pytorch官方
实现的FCN源码,官方源码地址:https://github.com/pytorch/vision/tree/main/torchvision/models/segmentation
1.1 环境配置
- Python3.6/3.7/3.8
- Pytorch1.10
- Ubuntu或Centos(Windows暂不支持多GPU训练)
- 最好使用GPU训练
- 详细环境配置见requirements.txt
1.2 文件结构
├── src: 模型的backbone以及FCN的搭建
├── train_utils: 训练、验证以及多GPU训练相关模块
├── my_dataset.py: 自定义dataset用于读取V