1.torchvision
PyTorch와 함께 사용되는 데이터 전처리 및 모델 구축을 위한 라이브러리
!pip install torchvision
1-1.transforms
데이터 전처리를 위한 다양한 변환 기능을 제공한다.
.Compose를 통해 여러 변환을 함께 사용할수있다.
transform = transforms.Compose([
transforms.Resize((224, 224)), # 이미지 크기 조정
transforms.ToTensor(), # 이미지를 텐서로 변환
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]) # 정규화
])
'AI' 카테고리의 다른 글
[LangChain] LangChain 이란 , 용어 (0) | 2025.01.10 |
---|---|
[AI] ONNX (Open Neural Network Exchange) (0) | 2025.01.09 |
[AI] Trainer API (0) | 2025.01.09 |