Skip to content

feat: improved model_exporter#608

Open
mgumowsk wants to merge 4 commits into
masterfrom
mgumowsk/fix-remaing-models
Open

feat: improved model_exporter#608
mgumowsk wants to merge 4 commits into
masterfrom
mgumowsk/fix-remaing-models

Conversation

@mgumowsk

@mgumowsk mgumowsk commented Jun 19, 2026

Copy link
Copy Markdown
Contributor

What does this PR do?

  • Fixed EfficientNet-V2-L preset: corrected mean/scale values from ImageNet standard (123.675 116.28 103.53 / 58.395 57.12 57.375) to 127.5 127.5 127.5 / 127.5 127.5 127.5
  • Fixed reverse_input_channels being set to false for one model preset — corrected to true
  • Fixed Mask R-CNN label mapping: replaced the naive labels shift with a proper COCO 91-class → COCO 80-class lookup table (COCO91_TO_COCO80) so category IDs that COCO dropped are handled correctly
  • Fixed Mask R-CNN preset: lowered confidence_threshold from 0.5 to 0.05 and changed labels from COCO_V1 to COCO_80
  • Added ExportParams frozen dataclass to pytorch.py to bundle export/quantization parameters instead of scattering individual config.get(...) calls
  • Extracted repeated boilerplate from all four converters (getitune, timm, torchvision, yolo) into shared BaseConverter helper methods

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

@mgumowsk mgumowsk requested a review from a team as a code owner June 19, 2026 08:22
@github-actions github-actions Bot added tests Related to tests python python related changes labels Jun 19, 2026
@mgumowsk mgumowsk added this pull request to the merge queue Jun 19, 2026
@github-merge-queue github-merge-queue Bot removed this pull request from the merge queue due to no response for status checks Jun 19, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

python python related changes tests Related to tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants