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

Instance Segmentation Update #58

Open
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

Mruzik1
Copy link

@Mruzik1 Mruzik1 commented Jun 15, 2023

  • Added a class for exporting YoloV5 instance segmentation model.
    • Inherits from YoloV5Exporter, overrides load_model and export_onnx, has a new method prune_model.
  • Updated the Exporter class.
    • Added a few changes to check what type of model we are exporting (methods get_onnx and export_openvino). Using the name is maybe not the best way to check the type. Although it has its own pros, plus we don't need to import anything that could affect flexibility of Exporter.
    • Changed quotation marks (' -> ") in the openvino command, because I had an error with those.
  • Added a little script test_seg_export.py + folder model_files for generated files to test the changes.

@tersekmatija
Copy link
Collaborator

Thanks! Let's get the #57 merged in first, then rebase and review @HonzaCuhel . In the meantime, let's get the ONNX and blobs shared internally so we can start integrating protos into the FW @Mruzik1 .

@HonzaCuhel
Copy link
Contributor

Sure! I want to first update the automatic version detection and then I'll merge it and deploy it. Is that okay, @tersekmatija ? Or should I merge the #57 now?

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

Successfully merging this pull request may close these issues.

3 participants