PGAN for small object detection
- 小目标检测的一个常用思路是提升图片输入分辨率,来增强小目标的分辨率和生成高分辨率的特征图,但这会导致训练和验证极度费时。
- 本文提出的PGAN方法对小目标生成高分辨率特征表示,使小目标的特征表示与大目标特征表示类似。
- 生成器网络通过较前层提取细粒度特征将小目标分辨率较低的特征转换为分辨率较高的特征。
- 判别器网络不仅用于生成小目标高分辨表示,同时证明带感知损失的生成高分辨女特征对检测准确率是有帮助的。
- 生成器网络被训练欺骗辨别器通过产生最像大目标表征的小目标,同时提升检测准确率。
- 辨别器被训练用于提升正确从实际大目标中分辨出生成的高分辨率表征,同时将定位准确率反馈给生成器。
-
小目标检测典型应用领域:交通标志检测、行人检测
模型架构.png
-
perception branch首先利用仅包含大目标的图片进行训练,然后利用仅包含小目标的图片进行训练,generator network 被训练用于对小目标生成高分辨率的类似大目标的表征。adversarial branch被训练用于区分生成的小目标高分辨率表征与实际大目标的原始表征。
模型细节.png
- generator 从底层提取细粒度特征,放入深度残差网络,通过element-wise加和操作,将深度残差网络学习到的残差特征与conv5学习到的特征结合,得到高分辨率特征图;discriminator将大目标和小目标的高分辨特征作为输入,adversarial分支 用于判断输入相片属于真的大目标的概率值,perception分支用于分类和边界框回归,验证从生成高分辨率特征获得的检测准确率提升。
网友评论