目标检测——夏侯南溪目标检测模型之数据结构

4.1 dets数据的一般表示——(list[list])

对于图像级数据的话,使用list会比较方便一些,因为图像的数量往往跟样本个数有关,

不容易直接确定下来,而list类型是一种动态的存储结构,可以使用list.append()方法来插入对象,还是挺方便的;

格式为:

[x0,y0,x1,y1,score] list

4.2 dets数据的高效表示——(multi_list)

这里的这个表示方法是借鉴了赵老师代码的表示方法,用

img_ids   list

dets         list

scores     list

三个平行的list形成的结构,可以直接转换为Tensor,还是挺方便的;

猜你喜欢

转载自blog.csdn.net/songyuc/article/details/106638618