[이론 정리] [torchvision] 자동으로 이미지 데이터셋 생성해주는 함수
2023. 3. 16. 13:37ㆍAI/이론 정리
torchvision.datasets.ImageFolder
root/
├── class1/
│ ├── img1.jpg
│ ├── img2.jpg
│ ├── ...
├── class2/
│ ├── img3.jpg
│ ├── img4.jpg
│ ├── ...
├── ...
import torchvision.datasets as datasets
# 이미지 데이터셋 경로 설정
data_path = 'root/'
# ImageFolder 클래스를 사용하여 데이터셋 생성
dataset = datasets.ImageFolder(data_path, transform=transforms.ToTensor())
(source: https://openai.com)
class별로 폴더를 나누고 하위 디렉토리에 img 파일들을 넣어주고 torchvision.datasets.ImageFolder 함수를 통해 dataset을 생성할 수 있다.
이 dataset[0]은 image 정보를 담고 있고 dataset[1]은 label 정보를 담고 있다.
'AI > 이론 정리' 카테고리의 다른 글
[이론 정리] CNN(Convolutional Neural Network)에 대하여 (0) | 2023.03.21 |
---|---|
[이론 정리] pytorch를 쉽게 사용할 수 있는 Tool(Pytorch Lightning) (0) | 2023.03.20 |
[이론 정리] Deep learning의 학습을 잘하기 위해서 알아두면 좋은 것 (0) | 2023.03.20 |
[이론 정리] Deep learning의 중요한 4 요소 (0) | 2023.03.20 |
[이론 정리] Pytorch (1) | 2023.03.13 |