안녕하세요. 저는 학교 과제로 딥러닝을 이용한 사진 분류 과제를 하고 있는 학생입니다. 저는 과제를 하기 위해 유튜브의 NeoWizard 님의 '[TensorFlow 2.x 강의 27] 나만의 이미지 시스템 구현' 영상을 보고 다람쥐와 청설모 사진을 분류하는 딥러닝 프로젝트를 진행하였습니다.

하지만 인덱싱 함수값이 KeyError가 계속 나서 과제를 끝내지 못한 상황입니다. 구글링을 통해 if문과 try-except문도 사용해 보았지만 배열이 아예 나오지 않아 다음 코드가 실행이 안되는 중입니다. 제발 저의 과제를 위해 많은 도움 부탁드립니다.

import random
import os
import numpy as np

import cv2

import glob

label_dict = {'chipmunk': 0, 'red': 1}

test_image_files_list = glob.glob(root_dir+ '/squirrel_image/test_image_files/*.jpg')

random.shuffle(test_image_files_list)

test_num = 16

test_image_files = test_image_files_list[:test_num]  # 태스트 파일이름은 정답.숫자.jpg 

label_list = []

for i in range(len(test_image_files)):
    label = test_image_files[i].split('/')[-1].split('.')[0].strip()
    label_list.append(label_dict[label])

src_img_list = []

for i in range(len(test_image_files)):
    src_img = cv2.imread(test_image_files[i], cv2.IMREAD_COLOR)
    src_img = cv2.resize(src_img, dsize=(IMG_WIDTH, IMG_HEIGHT))
    src_img = cv2.cvtColor(src_img, cv2.COLOR_BGR2RGB)
    src_img = src_img / 255.0

    src_img_list.append(src_img)

src_img_array = np.array(src_img_list)

label_array = np.array(label_list)

print(src_img_array.shape, label_array.shape)

print(label_array)