Инструмент создания реального синтетического набора данных
Этот инструмент используется для создания синтетических наборов данных с целью тестирования и сравнительного анализа алгоритмов машинного обучения. Инструмент написан на Python и использует библиотеку OpenCV для манипулирования данными и NumPy для числовых операций. Инструмент предназначен для использования в среде Python.
Вот понятный рабочий процесс инструмента.
Монтаж
Предварительные условия
Подготовьте данные и извлеките маску прозрачного изображения из этого репозитория Извлечение прозрачного изображения с использованием маски (Pipeline-1)
Требования
Для инструмента требуется Python 3.7 или выше.
pip install requirements.txt
Иерархия папок
├── Dataset │ │ ├── background_images │ │ ├── 1.jpg │ │ ├── 2.jpg │ │ ├── 3.jpg │ │ ├── input │ │ ├── 0-class-0 [folder name starting with class number] │ │ │ ├── 1.png │ │ │ ├── 2.png │ │ │ ├── 3.png │ │ ├── 1-class-1 │ │ │ ├── 1.png │ │ │ ├── 2.png │ │ │ ├── 3.png │ │ ├── 2-class-2 │ │ │ ├── 1.png │ │ │ ├── 2.png │ │ │ ├── 3.png │ │ ├── Dataset_output │ │ ├── 1.jpg │ │ ├── 1.txt │ │ ├── 2.jpg │ │ ├── 2.txt │ │ ├── 3.jpg │ │ ├── 3.txt
Применение
Инструмент предназначен для использования в среде Python.
python setup.py
Графический интерфейс выбора объекта
Объект выбран
Нажмите на базовое изображение.
Объект размещен
Автоматически создаваемые файлы этикеток в формате YOLO
Этот проект распространяется по лицензии Apache License 2.0 — подробности см. в файле ЛИЦЕНЗИЯ.
Первоначально опубликовано на https://github.com.