MobileNet1 torchvision.models에 있는 MobileNet V3 사용법 pyTorch 라이브러리에는 이미 사용할 수 있는 딥러닝 모델이 있습니다. 그 중에 가볍고 실시간으로 사용할만한 모델 중 하나인 MobileNet V3를 사용하는 방법에 대해서 소개 하도록 하겠습니다. 먼저 라이브러리를 import를 하도록 하겠습니다. import torch# 파이토치 from torchvision import transforms# 전처리 하기 위한 transfrom import torchvision.models as models# MobileNet V3이 포함되어 있는 models를 import from PIL import Image# 추론할 이미지를 읽어오기위한 Image를 import 그리고 MobileNet를 사용하기 앞서 GPU를 사용하기 위한 cuda 설정과 입력 이미지를 M.. 2023. 1. 12. 이전 1 다음