Skip to content

Дедубликатор изображений с единичным номером у транспортного средства (с использованием YOLO v8)

Notifications You must be signed in to change notification settings

AleksandrVechtomov/image-dedublicator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Как пользоваться

Настройте объект класса ImageDedublicator

Необходимо в переменную source_folder записать директорию к вашим изображениям (можно в подпапках)
Настроить атрибуты экземпляра класса ImageDedublicator:

  • указать в camera_id номер камеры (одну из 247, 248, 249, 252)

Метод process.run() запускает процесс отбора изображений.

source_folder = 'Source_images'

process = ImageDedublicator(source_dir=source_folder,
                            camera_id=248)

process.run()

Алгоритм работы программы:

  1. Пользователь указывает директорию к изображениям в переменной source_folder.
  2. Указывает номер камеры в атрибутах экземпляра класса ImageDedublicator.
  3. Запускает код программы main.py.
  4. Программа получает все пути к изображениям, указанным в source_folder.
  5. В цикле происходит обработка каждого изображения.
  6. В зависимости от номера камеры в camera_id применяется нужная зона интереса для каждого текущего изображения.
  7. В зоне интереса обнаруживается и распознаётся номер автомобиля.
  8. Распознанный номер сохраняется в список распознанных номеров.
  9. Если распознанного номера еще нет в списке распознанных номеров, то сохраняем изображение в виде вырезанного bbox автомобиля.
  10. В конце выводим сводную статистику по работе программы.

Исходные изображения (пример расположения):

Исходные изображения (в одной из папок):

Старт работы программы (в терминале):

Конец работы программы со статистикой (в терминале):

Выходные изображения (пример расположения):

Выходные изображения (в одной из папок):

Установка:

  1. Клонируйте репозиторий себе на компьютер
  2. Откройте папку проекта в любой IDE. Убедитесь, что в терминале вы находитесь в папке проекта
  3. Создайте виртуальное окружение
python -m venv venv
  1. Перейдите в виртуальное окружение и активируйте его
venv\Scripts\activate
  1. Установите все необходимые библиотеки
pip install -r requirements.txt
  1. Запустите код на выполнение
python main.py

About

Дедубликатор изображений с единичным номером у транспортного средства (с использованием YOLO v8)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages