Как визуализировать набор данных с помощью аннотации XML и использовать его для YOLOv5

Я работаю над индивидуальным обучением для YOLOv5. Я хочу обучить YOLOv5 с помощью наборов данных из UA-DETRAC, который содержит аннотацию XML для папка с изображениями. Как я могу визуализировать XML-аннотацию наборов данных, подобную этой

визуализация наборов данных

и как использовать его для кастомного обучения YOLOv5? Я прочитал это, но в этом наборе данных есть только 1 аннотация XML для каждой папки (которая содержит несколько изображений), а не 1 XML для 1 изображения.


person juliussin    schedule 03.11.2020    source источник


Ответы (1)


Вы можете визуализировать все свои записи и изображения с помощью labelimg https://github.com/tzutalin/labelImg

Чтобы преобразовать xml (формат паскаль / вокал) в текст (формат йоло), вы можете использовать программу labelimg, затем щелкнуть паскаль / вокал, и он изменится на формат йоло. Щелкните изображение проверки и сохраните. Программа сохранит в формате yolo изображение, на котором вы находитесь.

Для автоматизированного преобразования вы можете использовать этот пример в качестве шаблона для написания собственного конвертера https://github.com/rafael-junio/mio-dataset-converter/blob/master/mio-dataset-converter.py

person Rafael Junio Xavier    schedule 03.11.2020
comment
Спасибо! Попробую лейбл img. как насчет формата папки, которая содержит несколько изображений, при этом XML-файл только для папки, а не для всех изображений. - person juliussin; 04.11.2020
comment
В моем примере вы можете видеть, что я перебираю один CSV, вы можете сделать что-то подобное. - person Rafael Junio Xavier; 04.11.2020