Инструмент создания реального синтетического набора данных

Этот инструмент используется для создания синтетических наборов данных с целью тестирования и сравнительного анализа алгоритмов машинного обучения. Инструмент написан на 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.