EasyOCR 커스텀 데이터셋에서 finetuning 하기 (3) - recognition model train

https://github.com/JaidedAI/EasyOCR/blob/master/custom_model.md

EasyOCR/custom_model.md at master · JaidedAI/EasyOCR

위의 링크인 github에 어느 정도 자세히 설명해주고 있다.

---

나의 경우에는 일단 detection을 부분을 건들지 않고, recognition 부분만 finetuning하는 것을 목표로 세웠다.

(그림에서는 ResNet+LSTM+CTC 부분을 의미한다)

detection finetuning은 해보지 않았지만 깃허브에서 역시 하는 방법을 설명해주고 있다.

---

1. 데이터셋 준비

데이터셋들은 이미 준비가 되어 있었다 > 이미지와 labels.csv

lables.csv에는 아래와 같이 각각의 이미지에 대한 annotation된 정보가 들어있다.

example.png Hello Kitty

 

파일 구성은