我想使用视网膜网络训练自己的数据集。 Retinanet github页面说我需要将数据集的注释转换为csv格式。 Retinanet CSV格式:
path/to/image.jpg,x1,y1,x2,y2,class_name
我已经将自己数据集的注释转换为python列表。清单示例:
['/home/atakan/VOC2007/JPEGImages/bookstore_video0_1.jpg,1382,553,1420,602,Pedestrian',...]
将列表转换为CSV代码:
with open('output.csv', 'w', newline='') as csvfile:
writer = csv.writer(csvfile,delimiter =' ',quotechar =',',quoting=csv.QUOTE_MINIMAL)
for i in liste:
writer.writerows([[i]])
但是我无法获得retinanet格式的csv输出。我在哪里做错了?谢谢...