我想使用视网膜网络训练我自己的数据集。Retinanet github 页面说我需要将数据集的注释转换为 csv 格式。视网膜网 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]])
但我无法获得视网膜格式的 csv 输出。我在哪里做错了?谢谢...