Solución para problemas con la velocidad de inferencia de la red de detección

Solución para problemas con la velocidad de inferencia de la red de detección

de Mercedes Marzoa -
Número de respuestas: 1

Hola! 

Para quienes tengan problemas de velocidad de inferencia con la red de detección de la pelota de tenis en su máquina pueden bajar el tamaño de la imagen que usan para inferir agregando en esta linea de  el tamaño de la imagen:   

results = self.model(frame,conf=COFIDENCE_THRESHOLD, imgsz=SIZE_IMAGE)[0]

 Pueden probar con 320 o 160. 

Además entrenamos un modelo más chico que estamos subiendo ahora, eso también debería mejorarles el tiempo.

Saludos, 

Mercedes

En respuesta a Mercedes Marzoa

Re: Solución para problemas con la velocidad de inferencia de la red de detección

de Jorge Visca -
El modelo compacto que preparó Mercedes está disponible acá. Este modelo es útil si usan máquinas menos potentes, o simplemente prefieren ahorrar CPU. Pueden copiar el archivo a mano dentro de su instalación, o pueden modificar el Dockerfile cambiando la linea

ADD https://www.fing.edu.uy/~jvisca/fra/tennis0.pt /home/ubuntu/yolo_model/
 
por

ADD https://www.fing.edu.uy/~jvisca/fra/tennis_nano.pt /home/ubuntu/yolo_model/

Luego recuerden modificar tennis_ball_detector.py para cargar el modelo deseado.