SW 마에스트로 활동으로 작성된 코드입니다. 웹툰 영역 글자 인식에 맞는 학습 가능 모듈 및 경량화 모듈을 작성하였습니다.
train_model.py
내부의 파라미터 조정 후
python train_model.py
를 실행하여 모델 학습을 실시합니다.
predict.py
내부에 실제 예측을 위한 코드가 저장되어 있습니다. 이를 실행하여 예측이 가능합니다.
Components
에는 OCR에 필요한 각각의 모듈들이 저장되어 있습니다.
crnn_model
: 실제 사용된 모델 파일입니다.dataset
: 데이터 입력을 위한 dataset 클래스tokenizer
: 예측 결과값을 문장으로 바꿔 주는 클래스trainer
: 학습을 위한 코드들을 담고 있는 모듈
- 웹툰에서 추출된 웹툰 대사 이미지 100장으로 성능 테스트
- 파이썬 오픈소스 라이브러리인
tesseract
한국어 버전과 성능을 테스트함
자체 모델 | tesseract | |
---|---|---|
단어별 정확도(%) | 26% | 0% |
글자별 정확도(%) | 61% | 21% |