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

How to configure yolov5-seg? #146

Open Anleeno-Xu opened 9 months ago

Anleeno-Xu commented 9 months ago

I have made a yaml file and loaded a yolov5-seg model, but the labeling results are just boxes (just few images have results).

type: yolov5
name: yolov5n-seg
display_name: YOLOv5n-seg
model_path: y5n_seg.onnx
confidence_threshold: 0.25
input_height: 640
input_width: 640
nms_threshold: 0.45
score_threshold: 0.5
classes:
  - a
  - b
Anleeno-Xu commented 9 months ago

'type: yolov5_seg ' is also wrong.