딥러닝 모델을 학습시키기 위해서는 파일시스템에서 데이터셋을 가져와야 한다.
여러 개의 이미지로 구성된 데이터셋을 가져오려면 파일시스템에서
정해진 경로 내에 있는 모든 파일명을 읽어올 필요가 있다.
경로 내의 파일명을 가져오는 코드는 다음과 같다.
dir_name = "D:/Dataset/Training_Dataset/"
train_list = os.listdir(dir_name)'Python' 카테고리의 다른 글
| [PyTorch] Setting up PyTorch plugin "upfirdn2d_plugin"... 문제 (0) | 2022.12.20 |
|---|---|
| [Conda] 콘다 업데이트의 위험성.. / CondaSSLError: OpenSSL (0) | 2022.12.20 |
| [PyTorch] CUDA capability sm_86 is not compatible with the current PyTorch installation (0) | 2022.12.12 |
| [Tensorflow] Pycharm tensorflow Error (0) | 2022.05.25 |
| [Anaconda]sudo 명령어 없이 ffmpeg 설치 (0) | 2022.05.23 |
