Pytorch/Tutorials
Tutorial[8] - Inference(끝)
딥러닝의 이론적인 내용은 최대한 배제하고 Pytorch를 활용하여 코딩할 수 있도록 필수로 알아야하는 내용에 대해서만 초점을 두었습니다. Pytorch 1.4 버전을 기준으로 공식 홈페이지에 나와있는 튜토리얼을 참고하였으며, 데이터 로드, CNN모델 설계, 학습, 추론까지의 일련의 과정을 경험해보는것이 Tutorial의 목표입니다. Inference 이제 마지막으로 남은것은 추론이다. 앞서 학습을 통해 획득한 모델을 통해 새로운 이미지에 대하여 잘 분류를 해내는지 확인을 하는 작업을 수행한다. 프로세스 : import torch import MyNetwork import cv2 def classification_img(img_path, model): device = torch.device('cuda:0'..
2020. 10. 4. 21:56
최근댓글