LibTorch 강의 4. TorchScript로 모델 저장 및 불러오기 trace 및 Torch Script로 resnet18 모델 load 및 추론 imagenet label

지난 시간에 우리는 LibTorch를 이용하여 python 에서 학습한 모델을 로드하여 추론하는 방법에 대해 간단히 살펴보았습니다. https://blog.naver.com/dorergiverny/223918911614 LibTorch 강의 3. 모델 구현 및 학습하기 MNIST 데이터 다운로드 숫자 인식 신경망 구성 네트워크 python NLLLoss TorchScript 영상저장 OpenCV 지난 시간에 우리는 LibTorch의 기본 사용법을 정리하면서 Tensor를 생성하고 간단히 조작하는 방법에 ... blog.naver.com 이번에는 LibTorch 를 이용하여 PyTorch 로 학습된 모델을 읽어오고 C++에서 LibTorch를 이용하여 inference 하는 방법에 대해 좀 더 살펴보겠습니다.

LibTorch 에서 미리 학습된 모델을 로딩하고 사용하는 방법은 크게 두가지가 있습니다. PyTorch 로 학습 후 weight 를 Torch Script로 변환 후 LibTo...