[Python] easyocr 사용자 모델 학습하기 - 모델 학습

 

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

 

1. 프로젝트 설치

# 소스코드 내려받기
$ git clone https://github.com/clovaai/deep-text-recognition-benchmark.git

 

2. 학습 데이터 준비

이전 포스팅을 참고하여 준비

 

3. 학습 데이터를 lmdb 포맷으로 변환

python create_lmdb_dataset.py --inputPath ../TRDG2DTRB/output/ --gtFile ../TRDG2DTRB/output/gt.txt --outputPath data_lmdb_release/training/MJ/MJ_test
python create_lmdb_dataset.py --inputPath ../TRDG2DTRB/output/ --gtFile ../TRDG2DTRB/output/gt.txt --outputPath data_lmdb_release/training/MJ/MJ_train
python create_lmdb_dataset.py --inputPath ../TRDG2DTRB/output/ --gtFile ../TRDG2DTRB/output/gt.txt --outputPath data_lmdb_release/training/MJ/MJ_validation
python create_lmdb_dataset.py --inputPath ../TRDG2DTRB/output/ --gtFile ../TRDG2DTRB/output/gt.txt --outputPath data_lmdb_release/training/ST
python create_lmdb_dataset.py --inputPath ../TRDG2DTRB/output/ --gtFile ../TRDG2DTRB/output/gt.txt --outputPath data_lmdb_release/validation

 

구조

├─data_lmdb_relea

│ ├─training

│ │ ├─MJ

│ │ │ ├─MJ_test

│ │ │ │ ├─ data.mdb

│ │ │ │ ├─ lock.mdb

│ │ │ ├─MJ_train

│ │ │ │ ├─ data.mdb

│ │ │ │ ├─ lock.mdb

│ │ │ └─MJ_validation

│ │ │ │ ├─ data.mdb

│ │ │ │ ├─ lock.mdb

│ │ └─ST

│ │ │ ├─ data.mdb

│ │ │ ├─ lock.mdb

│ └─validation

│ │ ├─ data.mdb

│ │ ├─ lock.mdb

* 추후에는 폴더 별 목적에 맞는 파일이 필요

 

4. 기존에 이미 학습된 모델 준비

github.com/clovaai/deep-text-recognition-benchmark#run-demo-with-pretrained-model

 

해당 파일 다운로드

다운받은 파일 models 경로에 저장

5. demo.py 실행
demo.py를 실행하여 다운로드 받은 모델과 프로젝트의 정상 동작을 확인

python demo.py 
--Transformation None 
--FeatureExtraction VGG 
--SequenceModeling BiLSTM 
--Prediction CTC 
--image_folder demo_image/ 
--saved_model ./models/None-VGG-BiLSTM-CTC.pth

 

정상 동작 확인

6.모델 학습하기

* GPU가 탑재된 데스크탑환경에서 학습이 가능하다

* 학습 시 소스 코드 수정 필요(train.py)

1. 최상단에 # -*- coding: utf-8 -*- 추가 (syntax error 해결)

2. 
parser.add_argument('--data_filtering_off', action='store_true', help='for data_filtering_off mode')를
parser.add_argument('--data_filtering_off', action='store_true', default=True, help='for data_filtering_off mode')로 수정

3. 옵션 추가 (한글)
opt.character = '0123456789abcdefghijklmnopqrstuvwxyz가각간갇갈감갑값갓강갖같갚갛개객걀걔거걱건걷걸검겁것겉게겨격겪견결겹경곁계고곡곤곧골곰곱곳공과관광괜괴굉교구국군굳굴굵굶굽궁권귀귓규균귤그극근글긁금급긋긍기긴길김깅깊까깍깎깐깔깜깝깡깥깨꺼꺾껌껍껏껑께껴꼬꼭꼴꼼꼽꽂꽃꽉꽤꾸꾼꿀꿈뀌끄끈끊끌끓끔끗끝끼낌나낙낚난날낡남납낫낭낮낯낱낳내냄냇냉냐냥너넉넌널넓넘넣네넥넷녀녁년념녕노녹논놀놈농높놓놔뇌뇨누눈눕뉘뉴늄느늑는늘늙능늦늬니닐님다닥닦단닫달닭닮담답닷당닿대댁댐댓더덕던덜덟덤덥덧덩덮데델도독돈돌돕돗동돼되된두둑둘둠둡둥뒤뒷드득든듣들듬듭듯등디딩딪따딱딴딸땀땅때땜떠떡떤떨떻떼또똑뚜뚫뚱뛰뜨뜩뜯뜰뜻띄라락란람랍랑랗래랜램랫략량러럭런럴럼럽럿렁렇레렉렌려력련렬렵령례로록론롬롭롯료루룩룹룻뤄류륙률륭르른름릇릎리릭린림립릿링마막만많말맑맘맙맛망맞맡맣매맥맨맵맺머먹먼멀멈멋멍멎메멘멩며면멸명몇모목몬몰몸몹못몽묘무묵묶문묻물뭄뭇뭐뭘뭣므미민믿밀밉밌및밑바박밖반받발밝밟밤밥방밭배백뱀뱃뱉버번벌범법벗베벤벨벼벽변별볍병볕보복볶본볼봄봇봉뵈뵙부북분불붉붐붓붕붙뷰브븐블비빌빔빗빚빛빠빡빨빵빼뺏뺨뻐뻔뻗뼈뼉뽑뿌뿐쁘쁨사삭산살삶삼삿상새색샌생샤서석섞선설섬섭섯성세섹센셈셋셔션소속손솔솜솟송솥쇄쇠쇼수숙순숟술숨숫숭숲쉬쉰쉽슈스슨슬슴습슷승시식신싣실싫심십싯싱싶싸싹싼쌀쌍쌓써썩썰썹쎄쏘쏟쑤쓰쓴쓸씀씌씨씩씬씹씻아악안앉않알앓암압앗앙앞애액앨야약얀얄얇양얕얗얘어억언얹얻얼엄업없엇엉엊엌엎에엔엘여역연열엷염엽엿영옆예옛오옥온올옮옳옷옹와완왕왜왠외왼요욕용우욱운울움웃웅워원월웨웬위윗유육율으윽은을음응의이익인일읽잃임입잇있잊잎자작잔잖잘잠잡잣장잦재쟁쟤저적전절젊점접젓정젖제젠젯져조족존졸좀좁종좋좌죄주죽준줄줌줍중쥐즈즉즌즐즘증지직진질짐집짓징짙짚짜짝짧째쨌쩌쩍쩐쩔쩜쪽쫓쭈쭉찌찍찢차착찬찮찰참찻창찾채책챔챙처척천철첩첫청체쳐초촉촌촛총촬최추축춘출춤춥춧충취츠측츰층치칙친칠침칫칭카칸칼캄캐캠커컨컬컴컵컷케켓켜코콘콜콤콩쾌쿄쿠퀴크큰클큼키킬타탁탄탈탑탓탕태택탤터턱턴털텅테텍텔템토톤톨톱통퇴투툴툼퉁튀튜트특튼튿틀틈티틱팀팅파팎판팔팝패팩팬퍼퍽페펜펴편펼평폐포폭폰표푸푹풀품풍퓨프플픔피픽필핏핑하학한할함합항해핵핸햄햇행향허헌험헤헬혀현혈협형혜호혹혼홀홈홉홍화확환활황회획횟횡효후훈훌훔훨휘휴흉흐흑흔흘흙흡흥흩희흰히힘?!'

 

train.py 실행

python train.py 
--train_data data_lmdb_release/training 
--valid_data data_lmdb_release/validation 
--select_data MJ-ST 
--batch_ratio 0.5-0.5 
--Transformation None 
--FeatureExtraction VGG 
--SequenceModeling BiLSTM 
--Prediction CTC 
--workers 0 #Multi-GPU 환경으로 실행
--num_iter 10000 #반복회수

 

학습이 진행 중

학습 완료

 

학습 된 모델

정상적으로 학습이 완료되고 학습된 모델이 생성된 것을 확인할 수 있다.

 

 

참고사이트

https://beok.tistory.com/100?category=1263133