Skip to content

Instantly share code, notes, and snippets.

@RyanKor
Created September 24, 2021 09:56
Show Gist options
  • Select an option

  • Save RyanKor/1e775db563ba58f6ff0b953a3c65be19 to your computer and use it in GitHub Desktop.

Select an option

Save RyanKor/1e775db563ba58f6ff0b953a3c65be19 to your computer and use it in GitHub Desktop.
# train/test 경로에 따른 이미지 파일 확인
# training image 폴더명 : 파일명 형태로 정리
train_file = {}
val_file = os.listdir(validation_dir)
for folder in os.listdir(train_dir):
train_file[folder] = os.listdir(train_dir + folder)
print(len(os.listdir(validation_dir)))
print(os.listdir(train_dir))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment