0

root_folder_train = '/content/gdrive/My Drive/Deep Learning FYP/Code/train/' root_folder_test = '/content/gdrive/My Drive/Deep Learning FYP/Code/test/' emo_folder = os.listdir(root_folder_train) # emo_folder包含 7 个训练图像文件夹 emo_folder_test = os.listdir(root_folder_test) # emo_folder 包含 7 个测试图像文件夹

def load_images_from_folder(root_folder,emo_folder,all_images,all_labels):
  all_images = []
  all_labels = []
  for cls, fol in enumerate(emo_folder):
    print(cls,fol)
    all_images_per_emo = os.listdir(os.path.join(root_folder, fol))
  for im in all_images_per_emo[:10]:
    img = cv2.imread(os.path.join(root_folder, fol, im),cv2.IMREAD_GRAYSCALE)
    all_images.append(img)
    all_labels.append(cls)
  return all_images
  return all_labels
root_folder = 'directory/folderpath'

train_images = []
train_labels = []
train_data =load_images_from_folder(root_folder_train,emo_folders_train,train_images ,train_labels)
4

0 回答 0