Chapter04 - Implemetation custom dataset

[waverdeep]

@dIronmanb Custom dataset을 구현해보는 것도 좋을 것 같습니다.

참고자료

• https://pytorch.org/tutorials/beginner/data_loading_tutorial.html
• https://wikidocs.net/57165
• https://sanghyu.tistory.com/90

아래 자료를 참고해서

StartDeepLearningWithPytorch/Chapter04/src/dataloader/dataloader.py

Lines 32 to 34 in 9433630

	train_set = ImageFolder(train_dir, transform = train_transform)
	valid_set = ImageFolder(val_dir, transform = train_transform)
	test_set = ImageFolder(test_dir, transform = train_transform)

위 부분을 수정해보시는 것도 좋을 것 같습니다.