学习用于单次目标检测的空间融合
作者:刘松涛、黄帝、王云红
介绍
在这项工作中,我们为金字塔特征融合提出了一种新颖的数据驱动策略,称为自适应空间特征融合(ASFF)。 它学习在空间上过滤冲突信息以抑制不一致的方法,从而提高特征的尺度不变性,并引入几乎免费的推理开销。 有关更多详细信息,请参阅我们的。
更新:
YOLOX 来 ,快来用更强的YOLO吧!
添加 MobileNet V2!
之前的模型实际上都是用错误的锚点设置训练的,我们修复了 mobileNet 模型上的错误。
我们目前不支持 mobileNet V2 的 rfb、dropblock 和 Feature Adaption。
mobileNet 的 FP16 训练现在不起作用。 我没弄明白。
mobileNet 的 FP16 测试下降了大约 0.2 mAP。
添加一个 demo.py 文件
更快的 NMS(采用官方
评论0
最新资源