CarDetection

Определение количества автомобилей на видеозаписи с камеры наблюдения с использованием Python 3.6, OpenCV, YOLO , CUDA и Tensorflow-GPU.

CPU

Для запуска:

1. Скачать код на компьютер и перейти в папку CPU.
2. Создать папку input и поместить в нее видео с камеры наблюдения.
3. Создать папку ouput (там будет результат).
4. Загрузить веса и поместить их в папку parameters.
5. Запустить программу :

python main.py --input input\Example.mp4 --output output\Example_CarDetection.mp4 --yolo parameters

GPU

Для запуска:

0. Установить все нужные программы по гайду.
1. Скачать код на компьютер и перейти в папку GPU.
2. Создать папку input и поместить в нее видео с камеры наблюдения.
3. Создать папку ouput (там будет результат).
4. Скачать веса и недостающие параметры.
5. Запустить программу :

python main.py --input input\ExampleGPU.mp4 --output output\ExampleGPU.mp4

Credits

При реализации использовалась информация из следующих источников :

1. YunYang1994/tensorflow-yolov3
2. qqwweee/keras-yolo3
3. eriklindernoren/PyTorch-YOLOv3
4. pjreddie/darknet
5. dmlc/gluon-cv
6. wizyoung/YOLOv3_TensorFlow
7. abewley/sort