개발/AI,ML,ALGORITHM
MNIST 데이터셋 다운로드
SKY STORY
2023. 7. 19. 11:19
반응형
문자인식을 학습을 위해 MNIST 데이터셋을 다운로드 받아
'data'폴더에 저장하는 코드이다.
//
// MNIST 데이터셋 다운로드
//
// Created by netcanis on 2023/07/20.
//
import torch
from torchvision import datasets, transforms
# MNIST 데이터셋 다운로드
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize((0.5,), (0.5,))
])
train_dataset = datasets.MNIST(root='./data', train=True, download=True, transform=transform)
test_dataset = datasets.MNIST(root='./data', train=False, download=True, transform=transform)
저장 결과는 다음과 같다.
2023.07.19 - [AI] - MNIST 데이터셋 다운로드
2023.07.19 - [AI] - MNIST 데이터셋을 이미지 파일로 복원
2023.07.19 - [AI] - MNIST 데이터셋 로더
2023.07.19 - [AI] - MNIST 모델 테스터
2023.07.19 - [AI] - MINST - SVC(Support Vector Classifier)
2023.07.19 - [AI] - MNIST - RandomForestClassifier
2023.07.19 - [AI] - MNIST - Keras
2023.07.19 - [AI] - MNIST - TensorFlowLite
반응형