[MMOCR] 사용방법

https://github.com/open-mmlab/mmocr

GitHub - open-mmlab/mmocr: OpenMMLab Text Detection, Recognition and Understanding Toolbox

---

1 . 데이터 전환(준비)

 

기존 LMDB format 데이터 -> image + Label based txt 로 변환

단순 코드

[LMDB] lmdb file loading, reading

멀티 프로세스 방식

[LMDB] lmdb file loading , image-txt format / LMDB 형식 이미지 라벨 텍스트 변환 (멀티프로세스 방식)

---

 

2 . VOC 전환

mmocr/mmocr/models/textrecog/convertors/base.py 내 신규 dict_type 추가(DICT90을 원하는 형식의 Vocab으로 변경)

---

3 . 학습/ 평가

다큐먼트 참조

Welcome to MMOCR’s documentation! — MMOCR 0.3.0 documentation

---

+) Inference 때 Visualize

한글 라벨링 깨지는 문제 -> OpenCV putText 한글 인코딩 문제 --> PIL 함수 변환 필요(이미 MMOCR에 있음)

 

mmocr/core/visualize.py 내 is_contain_chinese 함수에서 다음과 같이 return을 True로 반환하게끔 수정,

612번 라인 내 font_path를 한글 폰트로 변경

 

---

+) Tensorboard 사용

/mmocr/configs/_base_/default_runtime.py

GitHub Issue

visualize loss curve · Issue #514 · open-mmlab/mmocr