PaddleOCR-ONNX-Sample

PaddleOCRのPythonでのONNX推論サンプルです。
PaddleOCRの推論スクリプトからPaddle依存処理とインポートを削除しています。

Model

以下リストのモデルを変換しています（蒸留モデルではないモデルを使用）

• OCR Model List（V3, updated on 2022.4.28）

変換自体を試したい方はColaboratoryなどでPaddleOCR-Convert2ONNX.ipynbを使用ください。

Requirement

requirements.txt を参照ください。

Demo

デモ(シンプルなアルファベット検出)の実行方法は以下です。

python demo_simple_ocr_en.py --image=sample.jpg

• --image
  OCR対象画像の指定
  デフォルト：sample.jpg
• --det_model
  テキスト検出モデルの指定
  デフォルト：./ppocr_onnx/model/det_model/en_PP-OCRv3_det_infer.onnx
• --rec_model
  テキスト認識モデルの指定
  デフォルト：./ppocr_onnx/model/rec_model/en_PP-OCRv3_rec_infer.onnx
• --rec_char_dict
  辞書データの指定
  デフォルト：./ppocr_onnx/ppocr/utils/dict/en_dict.txt
• --cls_model
  テキスト回転分類モデルの指定
  デフォルト：./ppocr_onnx/model/cls_model/ch_ppocr_mobile_v2.0_cls_infer.onnx
• --use_gpu
  GPU推論の利用
  デフォルト：指定なし

デモ(日本語検出＋表示)の実行方法は以下です。

python demo_draw_ocr_ja.py --image=sample.jpg

• --device
  カメラデバイス番号の指定
  デフォルト：0
• --movie
  動画ファイルの指定 ※指定時はカメラデバイスより優先
  デフォルト：指定なし
• --image
  画像ファイルの指定 ※指定時はカメラデバイスや動画より優先
  デフォルト：指定なし
• --width
  カメラキャプチャ時の横幅
  デフォルト：640
• --height
  カメラキャプチャ時の縦幅
  デフォルト：360
• --det_model
  テキスト検出モデルの指定
  デフォルト：./ppocr_onnx/model/det_model/en_PP-OCRv3_det_infer.onnx
• --rec_model
  テキスト認識モデルの指定
  デフォルト：./ppocr_onnx/model/rec_model/en_PP-OCRv3_rec_infer.onnx
• --rec_char_dict
  辞書データの指定
  デフォルト：./ppocr_onnx/ppocr/utils/dict/en_dict.txt
• --cls_model
  テキスト回転分類モデルの指定
  デフォルト：./ppocr_onnx/model/cls_model/ch_ppocr_mobile_v2.0_cls_infer.onnx
• --use_gpu
  GPU推論の利用
  デフォルト：指定なし

Reference

• PaddlePaddle/PaddleOCR

Author

高橋かずひと(https://twitter.com/KzhtTkhs)

License

PaddleOCR-ONNX-Sample is under Apache2.0 License.

License(Image)

サンプル画像はフリー素材ぱくたそ様の写真を利用しています。

License(Font)

日本語フォントはLINE Seed JPを利用しています。