luxonis / depthai-python

DepthAI Python Library
MIT License
353 stars 192 forks source link

YOLO example is not working with models taken from the Luxonis model-zoo #878

Open stwerner97 opened 1 year ago

stwerner97 commented 1 year ago

I am facing some issues getting the YOLO example to work with models taken from the Luxonis model-zoo, that is, blob files other than those supplied in the example.

I compiled the blobs with the blobconverter.

import blobconverter

model_path = blobconverter.from_zoo(name="yolov6t_coco_416x416", zoo_type="depthai", shaves=8)

The tiny_yolo.py script then throws the error:

Traceback (most recent call last):
  File "<my-path>/depthai-python/examples/Yolo/tiny_yolo.py", line 77, in <module>
    detectionNetwork.setBlobPath(nnPath)
IndexError: map::at

I've tested this with yolov6t_coco_416x416 and yolop_320x320. Any advice?

Erol444 commented 1 year ago

Hi @stwerner97 , Could you share the full script? Seems like there's something wrong with that (eg. you are saving path to model_path, but are using nnPath afterwards).