Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Trying to include custom trained YOLACT_Edge Model #192

Open
raviakash opened this issue Oct 21, 2024 · 0 comments
Open

Trying to include custom trained YOLACT_Edge Model #192

raviakash opened this issue Oct 21, 2024 · 0 comments

Comments

@raviakash
Copy link

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant