search

OSLL / sirius-cv

0 stars 0 forks source link

Скрипт для отбора удачных изображений для датасета #46

Open light5551 opened 3 years ago

light5551 commented 3 years ago

Написать питон-скрипт linear_markup.py, который получает в аргументах путь к видео с камеры, кол-во пропуска кадров, подключает одно из решений, которое было реализовано в 1 семестре, и реализовывает функционал:

На экран выводится кадр, обработанный подключенным решением, с размеченными bounding box-ами. По нажатию на 1 и 2 происходит переход к следующему кадру с учетом пропуска кадров.

1 - неудачный кадр, не берем в датасет 2 - удачный кадр, берем в датасет

Доп функционал

owlengineer commented 3 years ago

В первом приближении не фокусируемся на формате данных

SaviDan245 commented 3 years ago

План реализации работы над разработкой скрипта (см. роад мап)

Егор (@sibenshtern):

  1. Входные данные: путь к видео/папке (обязательный аргумент), путь к выходной папке (дополнительный аргумент);
  2. Настроить argparse;
  3. Следовать по роад мапу до пункта Detect & classify the road sign(s);
  4. Выходные данные: цикл for по элементам папки (если таковые имеются):
    • считанное изображение, если очередным элементом папки является изображение;
    • цикл for со считыванием (с помощью OpenCV) фреймов из видео, если обязательным аргументом являлся путь к видео, или, если обязательным аргументом являлся путь к папке, в которой очередным элементом является видео.

P.S. Желательно писать небольшие комментарии к коду.

Даниил (@SaviDan245):

  1. Входные данные: считанное изображение;
  2. Функция детекции знака(-ов) и разметки изображения;
  3. Все оставшиеся действия в роад мапе;
  4. Выходные данные: папка с двумя элементами внутри: JSON-файлом с разметкой и папка с изображениями для датасета.
sibenshtern commented 3 years ago

Фичи для реализации:

owlengineer commented 3 years ago
  1. Сохранять картинку без квадрата
  2. Поменять выходной формат (https://github.com/OSLL/sirius-cv/issues/57)
  3. Перенести код в 46 ветку
  4. README.md добавить раздел по запуску скрипта разметки
  5. https://github.com/OSLL/sirius-cv/issues/59
SaviDan245 commented 3 years ago

@sibenshtern:

owlengineer commented 3 years ago

@sibenshtern Нужно пофиксить генерацию точек, они в неправильном порядке находятся

owlengineer commented 3 years ago

@sibenshtern оставить в папке signs только: stop, t-intersection, r-inter, l-inter, no-way-left, traffic_light