SOLOv2:SOLOv2

时间:2024-05-12 13:30:15
【文件属性】:

文件名称:SOLOv2:SOLOv2

文件大小:15.87MB

文件格式:ZIP

更新时间:2024-05-12 13:30:15

pytorch instance-segmentation mmdetection Python

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个纪元)


网友评论