CVHub520 / X-AnyLabeling

Effortless data labeling with AI support from Segment Anything and other awesome models.
GNU General Public License v3.0
3.06k stars 346 forks source link

Error throwing when use custom yolov8 segment model. #371

Open doskid opened 2 months ago

doskid commented 2 months ago

Hello,

I'm attempting to utilize a custom YOLOv8 segmentation model for auto-labeling, but encountered an error in the program: "Error in predict_shapes: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()."

Could you please advise on how to resolve this issue? Thank you.

Here is my YAML file:

type: yolov8_seg
      name: best_segment3.onnx
      display_name: trolley_detect_segment2
      model_path: D:\pycharm project\yoloDemo\best_segment3.onnx
      nms_threshold: 0.45
      confidence_threshold: 0.25
      classes:
        - trolley
CVHub520 commented 2 months ago

Hello @doskid, could you please share with me the version of X-AnyLabeling that you are currently utilizing?

doskid commented 2 months ago

Hello @doskid, could you please share with me the version of X-AnyLabeling that you are currently utilizing?

Sure, I'm using version release v2.3.5.

jpointchoi commented 2 months ago

We are facing same issue!