Instalación YoloV4 [py]

Este tutorial está basado en un tutorial de Ritesh Kanjee y está alojado en su web.

Paso 0. Descargar el código del proyecto aquí.

O paso 1.

conda install -c anaconda git

git clone https://github.com/augmentedstartups/yolov4-custom-functions.git

Paso 2. Instalar el enviroment en Conda.

cd yolov4-custom-functions

Paso 3a. Preparamos para ejecutar en CPU.

conda env create -f conda-cpu.yml

conda activate yolov4-cpu

Paso 3b. Preparamos para ejecutar en GPU (Sí todo está instalado correctamente)

conda env create -f conda-gpu.yml

conda activate yolov4-gpu

Paso 4. Download de Weights.

Aquí el link

Luego hay que convertirlo a Tensor Flow format.

python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4

Paso 5. Run.

¿Tienes la webcam conectada?

python detect_video.py --weights ./checkpoints/yolov4-416 --size 416 --model yolov4 --video 0 --output ./detections/results.avi

Si todo va bien, que supongo que hay que instalar y arrancar algún paquete más, todo debería funcionar.

Documentación.

Una explicación interesante.