목록Vit (1)

RoBoLoG

[Pytorch, timm] Vision Transformer로 Custom Dataset 학습하기 (이미지 분류)

[Pytorch, timm] Vision Transformer로 Custom Dataset 학습하기 (이미지 분류) 1. VERSION python = 3.8.10 pytorch = 1.9.1 timm = 0.4.12 2. 데이터 dataset 폴더 안에 각 class에 대한 폴더가 있고, class 폴더 안에 사진 파일들이 있는 상황 dataset --class1 ----class1_1.jpg ----class1_2.jpg --class2 ----class2_1.jpg ----class2_2.jpg 3. 코드 import timm import torch import torchvision import torch.utils.data as data import torchvision.transforms as ..
Study/Artificial Intelligence 2024. 2. 21. 12:15
이전 Prev 1 Next 다음