[数据集]voc2coco及cocoapi使用
COCO
提供了一个完备的工具包 - cocoapi,能够满足实际开发过程中对于图像、类别、标注信息的提取
以下操作中仅涉及python
使用
介绍COCO
目标检测任务以及评价标准
COCO
将目标检测任务分为两个部分:
用于目标检测任务的训练/测试/验证数据集拥有超过20
万张图像,共80
个类别。其中,在训练和验证数据集上的标注超过了50
万个
介绍COCO
数据集以及标注文件格式。官网地址:COCO
COCO
是一个大规模的目标检测、分割和字幕数据集。包含了以下特征:
Object segmentation
)Recognition in context
)Superpixel stuff segmentation
)33
万张图像,其中超过20
万张已标注(330K images (>200K labeled)
)150
万个目标实例(1.5 million object instances
)80
个目标类别(80 object categories
)91 stuff categories
5
个字幕(5 captions per image
)25
万个行人关键点(250,000 people with keypoints
)研究lufficc/SSD源码时,发现其实现了一个目标检测训练框架,进行深入研究后整理成一个新的训练框架
阅读源码时发现torchvision
提供了许多边界框操作,包括
NMS
IoU
计算在线文档:torchvision.ops
源码地址:torchvision/ops/boxes.py
当前torchvision
版本:0.6.0a0+82fd1c8
之前学习IoU
的概念并且实现了预测框和对应真值边界框之间的计算 - [目标检测]IoU
不过预设的条件是每个真值边界框仅和单个预测边界框进行比对,参考box_utils.py计算每个真值边界框和每个预测边界框的IoU
之前在[数据集][PASCAL VOC]07+12中需要额外下载、解压数据集才能进一步实现VOC 07+12
的集合。今天发现了PyTorch
的voc.py
集成了2007
测试集,同时可以结合ConcatDataset一起使用
相关实现:zjykzj/vocdev
4
月份实现了YOLO_v1
算法,学习新的模型ResNet/GoogLeNet/SqueezeNet/DenseNet/SSD
,同时小结了评估标准,包括mAP/Flops/Params Size/FPS/Top-k accuracy
本周学习了如何寻找最优学习率/权重衰减,以及进行warmup+CosineAnnearling
加速模型训练,同时加强了数据预处理,包括随机擦除/颜色抖动/Ten Crops
测试等
本周继续SSD
算法的学习和实现,同时开始新模型和训练方法的学习。本月期待完成新工作的寻找