В данном репозитории содержится основной код, необходимый для работы консольного приложения для классификации (бинарной или мультиклассовой) объектов дорожного движения. Ниже описан процесс установки и развертывания приложения.
Для установки и развертывания приложения необходимо установить среду разработки PyCharm, после чего создать пустой проект и установить на него через терминал Jupyter Notebook (Подробнее на оф. сайте: https://jupyter.org/install)
Файл RoadObjectDetection.ipynb является блокнотом Jupyter Notebook, содержащим основной код программы. Его необходимо скопировать в папку, созданного проекта, после чего через терминал запустить Jupyter (команда jupyter notebook). Далее достаточно запустить первую ячейку, после чего будет произведена установка yolov5.
Файл detectBinary.py необходимо скопировать в папку yolov5, которая была создана после исполнения первой ячейки. Этот файл нужен для бинарной классификации объектов.
Также необходимо скачать файл весов для нейронной сети, которые должны быть размещены в одном каталоге с RoadObjectDetection.ipynb. Ссылка: https://drive.google.com/file/d/1ITRuTdejsv8cDXbiGgtE5lnGObtO9mHe/view?usp=sharing
Далее достаточно поменять переменную path в 5 строке второй ячейки, записав туда путь до своего проекта в виде строки и приложение готово к работе. Можно запустить исполнение всех ячеек, после чего под последней ячейкой появится консоль, в которой можно работать.