论文模型对比需要对之前模型复现 从0记录实现过程
基础配置:pytorch+cuda,系统:win, 模型:Deeplabv3+,数据集:Aeroscapes
目录
基础代码来源
代码链接:deeplabv3+,可以直接gitclone也可以下载zip然后解压。
训练
将aeroscape数据集改名为VOC2007
然后变一下结构放到VOCdevkit中,如图。
修改train.py中的num_classes为12,其他都没动即可训练。
预测
predict.py十分详细的中文注释,一键运行。