1

반려동물 피부 질환

AI 허브

데이터 준비하기 부분 - code 오타가 있는거 같습니다

yuri
2022.11.16 00:55
212

데이터 준비하기 부분에서..

transform_train, transform_test만 있고, 

transform_val은 선언되지 않았습니다. 

또한 아래와 같이 code 수정 또는 보완이 된다면 좋을거 같습니다. 

  • 원본

train = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
test = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
val  = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)

  • 수정필요

train = torchvision.datasets.ImageFolder(root='경로', transform=transform_train)
test = torchvision.datasets.ImageFolder(root='경로', transform=transform_test)
val  = torchvision.datasets.ImageFolder(root='경로', transform=transform_val)

1
1개의 댓글
로그인 후 이용해주세요!