deep2 전이학습(Transfer Learning) [수화 이미지 분류] (4) 1. Dataset & Data Loader(1) Custom Datasetimport torchvisionfrom PIL import Imageclass Custom_dataset(torch.utils.data.Dataset): def __init__(self, paths, split=None, transform=None): self.paths = paths self.split = split self.transform = transform self.image = [] self.label = [] if self.split.upper() == 'TRAIN' or self.split.upper() =.. 2024. 6. 9. 전이학습(Transfer Learning) [개, 고양이 분류] (3) 1. Label Maplabel_map = { 'cat' : 0, 'dog' : 1} 2. Dataset & Data Loader(1) Custom Datasetclass CustomDataset(torch.utils.data.Dataset): def __init__(self, root_path, split, transform, label_map): self.split = split.upper() self.root_path = root_path self.transform = transform self.label_map = label_map self.image = [] self.label = [] .. 2024. 6. 9. 이전 1 다음