论文名称:《SOLOv2: Dynamic, Faster and Stronger》
论文链接:https://arxiv.org/abs/2003.10152
参考代码:http://github.com/WXinlong/SOLO (v1已开源~)
目录
综述
SOLO v1回顾
整体思路
操作流程(附源代码)
改进空间
SOLO v2
mask learning
Mask kernel branch
Mask feature branch
其他
mask NMS
实验结果
总结
综述
SOLO v2遵循了SOLO的优雅、简单的设计,并且针对mask的检测效果和运行效率做了两个改进:(1)mask learning:能够更好地学习到mask(2)mask NMS:提出了matrix nms,大大减少了前向推理的时间。 最终,SOLOv2相比SOLOv1,AP提升1.9%的同时速度快了33%,light-weight 版本的SOLVv2能够在COCO上达到31.3FPS,37.1%AP。除此之外,SOLO v2还在目标检测和全景分割任务中表现上佳,证实了SOLO的思路有用在更多视觉任务的潜力。