Pytorch/Tutorials
Tutorial[3] - DataSet(ImageFolder)
딥러닝의 이론적인 내용은 최대한 배제하고 Pytorch를 활용하여 코딩할 수 있도록 필수로 알아야하는 내용에 대해서만 초점을 두었습니다. Pytorch 1.4 버전을 기준으로 공식 홈페이지에 나와있는 튜토리얼을 참고하였으며, 데이터 로드, CNN모델 설계, 학습, 추론까지의 일련의 과정을 경험해보는것이 Tutorial의 목표입니다. Dataset 수집한 데이터를 불러오는 작업을 해야한다. Classification에서는 torchvision에서 제공하는 ImageFolder를 사용하는것이 가장 효율적이다. ImageFolder 사용법 : ImageFolder를 사용하기 위해선 가장먼저 수집된 데이터의 폴더구조를 아래와 같이 설계해야한다. 반.드.시. root_dir | --- cat/ | |-- 0001..
2020. 9. 30. 05:23
최근댓글