anylabeling
anylabeling copied to clipboard
error loading custom yolov5 model
Following https://anylabeling.nrl.ai/docs/custom-models for a custom yolov5 model results in
type: yolov5 name: Custom display_name: Custom model_path: ./best.onnx confidence_threshold: 0.45 input_height: 640 input_width: 640 nms_threshold: 0.45 score_threshold: 0.5 classes:
- A
- B
- C
- D
- E
- F
- G Any tips what the issue is?
Check this opencv issue here is the main reason for this error. It seems you need to convert to your trained model to ONNX format with opset =12.It worked for me and the model loaded successfully.