<计算机视觉 五> 模型训练时候标签数据的变换
在训练的时候pytorch Dataset模块会批量生成训练数据集,这里分析数据转换的过程。数据管道不断的循环遍历输出标注的数据,在代码中数据一个张图像中可能有很多个目标,一个训练的targets又可能由多个图像组成,一般一个batch有10张图像我们就称batch为10。图中第一组数据[0,3,0.10,0.30,0.45,0.19]的数据是,batch为0(即第一张图像),label为3,分类标签为第4个类别,xywh就是这个分类3的标签目标框所在地额中心和宽高,数据下方也给出了目标的框的详解。...
复制链接