Я хочу тренировать свой собственный набор данных, используя ретинанет.GitHub Retinanet говорит, что мне нужно преобразовать аннотации моего набора данных в формат 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]])
Но я не могу получить вывод csv в формате retinanet.Где я делаю не так? Спасибо ...