EasyOCR lmdb.Error: /home/data/train/: No such file or directory

hyereen·2023년 1월 7일

EasyOCR Finetuning을 위해 재학습할 때 나타난 에러이다. 분명 train 경로도 다 맞고 다른 파라미터들도 맞게 한 것 같은데... 아래와 같은 에러가 나타났다.

에러메시지

lmdb.Error: /home/data/train/: No such file or directory 

이유는 EasyOCR의 config yml 파일의 dataset 이름이 lmdb인데, lmdb가 아니고 simple data를 넣으려고 해서 생긴 일이다.
Easy OCR은 simple data를 변환해줘서 lmdb형태의 데이터를 넣어줘야 학습할 수 있다.

  • simpledataset: 폴더 안에 이미지 파일들이 있고, gt(label)파일에는 폴더 경로와 그 이미지의 ground truth 데이터가 적혀있음
    ex. img_01.png, img_02.png, img_03.png, ...
    gt.txt
    img_01.png apple
    img_02.png cherry
    img_03.png banana

  • lmdb: 위의 simpledataset 데이터를 create_lmdb_dataset.py을 이용해서 변환한 상태

해결 방법

  1. create_lmdb_dataset.py을 이용해서 simple dataset을 lmdb형태로 바꿔준다

  2. lmdb가 저장된 경로를 config.yml 파일의 학습 데이터셋 경로에 넣어준다

  3. 재학습

0개의 댓글