Skip to content

feat: improved model_exporter#608

Merged
mgumowsk merged 4 commits into
masterfrom
mgumowsk/fix-remaing-models
Jun 22, 2026
Merged

feat: improved model_exporter#608
mgumowsk merged 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
@mgumowsk
mgumowsk added this pull request to the merge queue Jun 22, 2026
@github-merge-queue
github-merge-queue Bot removed this pull request from the merge queue due to failed status checks Jun 22, 2026
@mgumowsk
mgumowsk added this pull request to the merge queue Jun 22, 2026
Merged via the queue into master with commit 3d7b9e6 Jun 22, 2026
33 checks passed
@mgumowsk
mgumowsk deleted the mgumowsk/fix-remaing-models branch June 22, 2026 06:59
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