AI Gravity

AI Framework/PyTorch 검색 결과

해당 글 2건

U-net

https://89douner.tistory.com/298 코드를 사용하면서 알게된 사실인데 이 모델의 Baseline은 u-net이지만 논문과 동일하지 않았다. 좀더 찾아본 결과 오히려 SegNet과 비슷한 느낌이 있다. 효율이 좋은 배치 노말이 역시 추가되었고, same padding에 U-net의 가장 아래쪽에서 오른쪽 Conv에서 1024가 아닌 512로 설정되어 있는 등 U-net과 약간의 차이가 있었다. 모델의 성능은 IOU 약 91% +-0.5% 를 웃돌았다. 다음 코드는 pytorch로 사용한 모델과 train 코드이다. ## 라이브러리 불러오기 import os import numpy as np import torch import torch.nn as nn from torch.utils...

AI Framework/PyTorch 2022. 9. 1. 19:31

LeNet-5

https://towardsdatascience.com/implementing-yann-lecuns-lenet-5-in-pytorch-5e05a0911320 Implementing Yann LeCun’s LeNet-5 in PyTorch Recently, I watched the Data Science Pioneers movie by Dataiku, in which several data scientists talked about their jobs and how they… towardsdatascience.com from google.colab import drive drive.mount('LeNet-5') import numpy as np from datetime import datetime impo..

AI Framework/PyTorch 2021. 11. 18. 16:44