[Python] easyocr 사용자 모델 학습하기 - 학습데이터 변환

easyocr 사용자 모델 학습하기 - 학습데이터 변환

---

지난 포스팅에서 학습데이터 생성하는 법을 다뤘다.

이번 포스팅에서는 학습데이터를 변환하는 법을 다뤄보도록 할 것이다.

 

1. 프로젝트 설치

# 소스코드 내려받기
$ git clone https://github.com/DaveLogs/TRDG2DTRB.git

 

2. 학습 데이터변환

#input_path => 기존 학습 데이터 경로(변환할 데이터가 있는 경로)
#output_path => 결과 폴더 경로(없을 시 자동 생성)
python convert.py --input_path ../TextRecognitionDataGenerator/out --output_path ./output 

3. 결과

python convert.py --input_path ../TextRecognitionDataGenerator/out --output_path ./output
input path:  /app/easyocr_learning/TextRecognitionDataGenerator/out
output path:  /app/easyocr_learning/TRDG2DTRB/output
Total file count:  50
50 / 50 Processing !!
Conversion complete!

 

결과 데이터 구조

-output
  -images
  -gt.txt

images
-image_00.jpg
-image_01.jpg
-image_02.jpg
...
-image_49.jpg

gt.txt
images/image_00.jpg 가
images/image_01.jpg 가게
images/image_02.jpg 가격
...
images/image_49.jpg 가이드

학습 데이터

위와 같이 학습 데이터가 정상적으로 변환된 것을 확인할 수 있다.

참고 사이트 : https://davelogs.tistory.com/80