Модуль для распознавания компонентов на платах.
Краткая инструкция по установке и запуску модуля detection. Описание содержания модуля смотрите в doc/readme.md.
Для работы модуля требуется Python 3.6.8.
-
Установите зависимости:
python -m pip install --upgrade pip python -m pip install -r requirements.txt -
Запустите пример:
python -m detection --image tests/elm_test1/image.png --draw-elements --save-json-resultМодуль возьмет изображение
image.png, распознает на нем элементы PCB, выведет их в консоль, а также создаст папкуlog, в которую положит распознанную картинку и файл с элементами.
python -m unittest discover tests
Оценка точности классификатора производится по формуле: (найдено элементов) / (всего на размеченной плате + не верно найденные).
Чтобы сгенерировать документацию, выполните действия:
python -m pip install pdoc3
python -m pdoc --html detection
Визуализатор рисует компоненты и пины для заданного классификатора. Чтобы запустить визуализатор, выполните команду:
python tools\visualizer.py путь_до_dump_файла_классификатора путь_до_папки_куда_сохранить_изображения
Классификаторы сохранены в dump-файлах в папке detection/dupms:
- в файле
clf_types.dumpсохранен классификатор, который тренировался на данных из файловsizes.csv,types.csvи датасета, который утерян. - в файле
clf_types_2.dumpсохранен классификатор, который получен из классификатораclf_types.dumpсмещением на 3% пинов некоторых SMD-компонентов. Подробности в задаче #99598.