Pytorch/Tutorials
Tutorial[4] - DataSet(CumstomDataset)
딥러닝의 이론적인 내용은 최대한 배제하고 Pytorch를 활용하여 코딩할 수 있도록 필수로 알아야하는 내용에 대해서만 초점을 두었습니다. Pytorch 1.4 버전을 기준으로 공식 홈페이지에 나와있는 튜토리얼을 참고하였으며, 데이터 로드, CNN모델 설계, 학습, 추론까지의 일련의 과정을 경험해보는것이 Tutorial의 목표입니다. CustomDataset 이미지를 수집할 때, 간혹 ImageFolder에서 사용할 수 있는 폴더구조를 갖지 못하는경우가 발생한다. 예를들어 아래와 같은 폴더 구조에서는 사용할 수 없다. root_dir/ | --- cat_0001.jpg | --- cat_0002.jpg | --- dog_0001.jpg | --- rabbit_0001.jpg | --- rabbit_0002..
2020. 10. 1. 11:41
최근댓글