上传者: 42166918
|
上传时间: 2021-11-15 09:45:27
|
文件大小: 15.87MB
|
文件类型: -
SOLOv2
该代码是SOLOv2的非官方pytorch实现
安装
请检查以获取安装说明。
训练
遵循与SOLOv1相同的方式。
单个GPU:
python tools/train.py configs/solov2/solov2_r101_3x.py
多GPU(例如8):
./tools/dist_train.sh configs/solov2/solov2_r101_3x.py 8
重物
可以在下载经过训练的模型码:qw4e
结果
在使用resnet-101主干在coco数据集上训练了36个纪元(3x)之后,在COCO test-dev2017数据集上的mAP为39.5。 在原始论文中,该模型在72个时代(6x)之后达到了39.7。
可视化(1个纪元)