Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

EasyOCRで追加学習させたモデルの作成および実行 #76

Open
YushinOgasawara opened this issue Aug 31, 2024 · 0 comments
Open

Comments

@YushinOgasawara
Copy link
Contributor

開発目的

オリジナルで提供されているEasyOCRでは精度の向上が見込めないため,追加学習させる.

考えられる開発内容

1. データセットの準備

独自のデータセットを準備.各画像に対応するラベル(テキスト)を用意.

2. データローダーの作成

画像とラベルを読み込み,PyTorchのDataLoaderでバッチ処理を行う.

3. CRNNモデルの設定

EasyOCRのCRNNモデルをロードし,トレーニング用にカスタマイズ.

4. 損失関数と最適化

損失関数(例:CTCLoss)と最適化手法(例:Adam)を設定

5. トレーニングループ

モデルをトレーニングして保存.

6. モデルの使用

保存したモデルをロードし,OCRに使用.

7. EasyOCRでカスタムモデルを使用

カスタムOCRクラスを作成して,学習済みモデルを使用.

考えられる開発時間

備考

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant