DeepLearning/pyTorch2 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. pytorch를 통한 xor 게이트 학습 예제 코드 : https://github.com/MainForm/xorNet 코딩을 처음 시작할때 "hello world"를 출력하는 것처럼 딥 러닝을 학습할 때 가장 처음으로 접하는 것이 XOR를 학습하는 것이라고 생각합니다. 기본적인 AND 연산이나 OR 연산의 경우 간단히 직선으로 구분하여 문제를 해결 가능합니다. 하지만 XOR의 경우에는 단순한 직선으로는 문제를 해결하지 못하고 곡선을 통해 문제를 해결하여야 합니다. 코딩의 순서로 Dataset를 만들고 Model를 만들고 나서 학습을 진행하도록 하겠습니다. Dataset.py import torch from torch.utils.data import Dataset from random import randint class xorDataSe.. 2022. 8. 27. 이전 1 다음
