search

vietanhdev / anylabeling

Effortless AI-assisted data labeling with AI support from YOLO, Segment Anything, MobileSAM!!
https://anylabeling.nrl.ai
GNU General Public License v3.0
2k stars 218 forks source link

error loading custom yolov5 model #143

Open svanbodegraven opened 10 months ago

svanbodegraven commented 10 months ago

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: