EmilioMachado
In general, mobilenetSSD has much better performance, but we have a Pytorch compatible version in the PR here: luxonis/depthai-ml-training58. Note that it is still being improved so I can't help with any bad performance.
What you are seeing with YoloV5 seems like wrong predictions though. When you use a model trained in our notebook, you should do the following:
1/ Take the best.pt and upload it to tools.luxonis.com
2/ Extract the downloaded .blob and .json
3/ Use this script with flags -m
(path to the blob) and -c
(path to the config)
If you do that, can you confirm this working for you? Do detections look OK? If yes, then everything is fine and there might be some issue in how you use SpatialLocationCalculator.