[数据集][COCO]目标检测任务评估
Image Localization Dataset是一个目标检测数据集,其标注文件使用了PASCAL VOC定义的格式,将其转换成COCO格式
[数据集][COCO]目标检测任务
[数据集]COCO简介
介绍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源码时,发现其实现了一个目标检测训练框架,进行深入研究后整理成一个新的训练框架
[目标检测][PyTorch]边界框操作
阅读源码时发现torchvision提供了许多边界框操作,包括
NMS- 移除小边界框
- 修剪边界框坐标,保证边界框坐标位于图像内
- 计算边界框面积
IoU计算
在线文档:torchvision.ops
源码地址:torchvision/ops/boxes.py
当前torchvision版本:0.6.0a0+82fd1c8
[目标检测]IoU计算
之前学习IoU的概念并且实现了预测框和对应真值边界框之间的计算 - [目标检测]IoU
不过预设的条件是每个真值边界框仅和单个预测边界框进行比对,参考box_utils.py计算每个真值边界框和每个预测边界框的IoU