anylabeling icon indicating copy to clipboard operation
anylabeling copied to clipboard

Trying to include custom trained YOLACT_Edge Model

Open raviakash opened this issue 1 year ago • 0 comments

I have an YOLACT Model (Yolact_Edge not converted to trt) which was trained on custom dataset. I have performed the ONNX conversion of the model and also prepared my config.yaml which includes the path to the model. I also included the model in the models.yaml file. THe ONNX file is store in offline as a result there is no download URL in the models.yaml file.

This is how my config.yaml looks like
config.yaml

classes:

  • paint_defect
  • color_defect confidence_threshold: 0.45 config_file: /home/anylabeling_data/models/YolactEdge/config.yaml display_name: YolactEdge has_downloaded: false input_height: 512 input_width: 512 model_path: TestCase_112_33325.onnx name: YolactEdge nms_threshold: 0.45 score_threshold: 0.5 type: yolact_edge

models.yaml

  • name: "yolov8x-r20230415" display_name: YOLOv8x download_url: https://github.com/vietanhdev/anylabeling-assets/releases/download/v0.4.0/yolov8x-r20230415.zip
  • name: "YolactEdge" display_name: YolactEdge

Even after doing this I am not able to see my model in the UI. It throws me an error message saying "Error in loading custom model. Invalid config file format. Please help me out on how to include Yolact_Edge model

raviakash avatar Oct 21 '24 12:10 raviakash