SOLOv2训练自己数据集(实例分割,停车位检测)
SOLOv2训练自己数据集(实例分割,停车位检测)
1. 环境搭建2. 修改配置文件3. 下载权重4. 自定义数据集转COCO格式4. 训练&测试5.实验结果
论文链接:https://arxiv.org/abs/2003.10152 代码链接:https://github.com/WXinlong/SOLO
1. 环境搭建
除了mmcv单独安装,其他用:
pip3 install requirements.txt
安装mmcv:solo的代码用mmdet v1.0.0,搭配的是mmcv 0.2.16(有人用将mmdet替换成openlab-mmdetection官网最新mmcv,希望这样可以更换mmcv为最新的,是不可行的!所以这样就必须用mmcv 0.2.16了)
用下面的方式会报错:
pip install mmcv-full==0.2.16 -f htt
共有 0 条评论