doctr
doctr copied to clipboard
[models] Ensure all TensorFlow models are ONNX exportable
Most users of the library are more interested in existing pretrained models to use for inference rather than training. For this reason, it's important to ensure we can easily export those trained models.
-
General
- [x] Add a unittest/CI job to check that all models can be exported with ONNX (#976, #979, #980)
-
Classification
- [x] VGG (#910)
- [x] ResNet (#910)
- [ ] MobileNet V3 (#978)
- [x] MAGC ResNet (#910)
-
Text Detection
- [x] LinkNet #979
- [x] DBNet #979
-
Text Recognition
- [x] CRNN #980
- [x] SAR #980
- [x] MASTER #980
@frgfm update to Onnx i think that's ok !? :)
If that's the case, would you mind linking all the PRs to the issue description? :pray:
Closed with #1182