anylabeling icon indicating copy to clipboard operation
anylabeling copied to clipboard

error loading custom yolov5 model

Open svanbodegraven opened this issue 2 years ago • 1 comments

Following https://anylabeling.nrl.ai/docs/custom-models for a custom yolov5 model results in image

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?

svanbodegraven avatar Sep 04 '23 09:09 svanbodegraven

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.

raviakash avatar Oct 23 '24 13:10 raviakash