배경
AI-HUB의 한국 수어 영상 데이터셋을 통해 학습된 KSL Model을
지금 진행 중인 프로젝트에 적용하여 API로 구현하기 위해서
해당 모델을 실행하기 위해 코랩 환경에서 의존성 설정을 모두 마치고 실행을 했다.
이런 오류가 발생해서 이유를 찾아보니
config 파일에서 데이터셋의 키 값의 오류라고 판단되었다.
해당 config 파일을 살펴보니
DATASET:
DATA_ROOT: "/home/storage/sde_2T"
TRAIN:
IMG_PREFIX: "NIA_sen"
ANN_FILE: "KSL_TRAIN_QD_final.csv"
VAL:
IMG_PREFIX: "NIA_sen"
ANN_FILE: "KSL_VAL_QD_final.csv"
WORKER_PER_GPU: 16
TRANSFORM:
RESIZE_IMG: (256, 256)
TEMPORAL_SCALING: 1.0
CROP_SIZE: (224, 224)
TEMPORAL_CROP_RATIO: 0.0
MEAN: (0.0637, 0.0988, 0.2312)
STD: (0.0643, 0.0556, 0.1150)
VOCABULARY:
EXCLUDE_TOKENS:
- __ON__ #not used in ksl
- __OFF__
이런 부분이 있었다.
하지만 이 부분에서의 문제점은 찾지 못했고,
main.py 코드를 통해 문제가 되는 부분을 찾아가 보았다.
main.py의 line 31에서 오류가 뜨고, yacs 라이브러리에서 오류가 뜨니
해당 라이브러리를 사용하는 get_cfg()를 확인했다.
해당 함수는 config/defaults.py 파일에서의 _C를 클론 해서 리턴한다.
config/defaults.py 파일을 보니 문제가 되는 키값이 주석처리로 되어있어서 풀어주었더니
위 키값 오류는 사라졌다.
하지만
이런 오류가 발생했다.
아직 DATA_ROOT와 TRAIN, VAL 부분을 바꾸어줘야 한다....
다음 포스트!
'AI' 카테고리의 다른 글
[AI] OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized (0) | 2022.12.09 |
---|