anylabeling
anylabeling copied to clipboard
Trying to include custom trained YOLACT_Edge Model
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