Detección de productos en estantes YOLOv8

Para sumergirte en la hechicería minorista con el modelo de detección de objetos en estantes YOLOv8, sigue estos pasos encantados:

pip install ultralyticsplus==0.0.28 ultralytics==8.0.43

Invoca el modelo y descubre sus secretos:

from ultralyticsplus import YOLO, render_result
import cv2

# Lanza un hechizo para invocar el modelo
model = YOLO('foduucom/shelf-object-detection-yolov8')

# Ajusta los parámetros mágicos
model.overrides['conf'] = 0.25 # Umbral de confianza NMS
model.overrides['iou'] = 0.45 # Umbral de IoU NMS
model.overrides['agnostic_nms'] = False # NMS indiferente a la clase
model.overrides['max_det'] = 1000 # número máximo de detecciones por imagen

# Establece la imagen
date_path = "path/to/your/shelf/image"

# Puedes pasar transmisión en vivo de la cámara o video
# Inicia el viaje místico a través de los cuadros de video
# (Recuerda tener tu tapiz minorista listo)
while cap.isOpened():
# Leer un cuadro del video
success, frame = cap.read()

if success:
# Desata la magia del YOLOv8
results = model(frame)

# Muestra la magia en el cuadro
annotated_frame = results[0].plot()

# Presenta el cuadro encantado
cv2.imshow("Hechicería Minorista YOLOv8", annotated_frame)

# Disipa el hechizo si se presiona 'q'
if cv2.waitKey(1) & 0xFF == ord("q"):
break
else:
# Vuelve a la realidad cuando el video termina
break

# Suelta el video cautivo y cierra el portal
cap.release()
cv2.destroyAllWindows()