文章预览
点击上方 “ 小白学视觉 ”，选择加" 星标 "或“ 置顶 ” 重磅干货，第一时间送达 转自 | 机器学习AI算法工程 使用的VOC数据集链接开放在文章中，预训练模型已上传Github，环境我使用 Colab pro ，大家下载模型做预测即可。 代码链接： https://github.com/lixiang007666/segmentation-learning-experiment-pytorch 使用方法： 下载VOC数据集，将 JPEGImages SegmentationClass 两个文件夹放入到data文件夹下。 终端切换到目标目录，运行 python train.py -h 查看训练 (torch) qust116-jq@qustx-X299-WU8:~/语义分割$ python train.py -h usage: train.py [-h] [-m {Unet,FCN,Deeplab}] [-g GPU] choose the model optional arguments: -h, -- help show this help message and exit -m {Unet,FCN,Deeplab}, --model {Unet,FCN,Deeplab} 输入模型名字 -g GPU, --gpu GPU 输入所需GPU
………………………………
