doctr icon indicating copy to clipboard operation
doctr copied to clipboard

[models] Ensure all TensorFlow models are ONNX exportable

Open fg-mindee opened this issue 4 years ago • 2 comments

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

fg-mindee avatar Jan 10 '22 11:01 fg-mindee

@frgfm update to Onnx i think that's ok !? :)

felixdittrich92 avatar Jul 07 '22 07:07 felixdittrich92

If that's the case, would you mind linking all the PRs to the issue description? :pray:

frgfm avatar Jul 07 '22 09:07 frgfm

Closed with #1182

felixdittrich92 avatar Apr 29 '23 13:04 felixdittrich92