Dickson Neoh
Dickson Neoh
The last row/column is the "nothing" class where the object pertaining to the class is not detected https://github.com/svpino/tf_object_detection_cm/issues/18
Anyone found a workaround?
This depends on the team behind DeepSparse. Have you tried asking in their Slack?
@lucheng07082221 did you complete the training? Or did you halt the training halfway?
Did you try to run my example on Colab? As far as I can tell it works
I would suggest using the codes I provided and trying them out on your dataset.. I could not replicate the issue you said above so its hard to troubleshoot
@abirsince92 have you tried running it on Colab? How do I produce this error on my machine?
When you train the model, what do you have in the .yaml file? For example, here's mine https://github.com/dnth/yolov5-deepsparse-blogpost/blob/main/yolov5-train/data/pistols.yaml
In that case you might want to change this line https://github.com/dnth/yolov5-deepsparse-blogpost/blob/dbd4306fb036a7f47b42937196d23bdbc953686a/yolov5-train/deepsparse_utils.py#L577
What about the older version on yolov5n ? Can you export those?