Tensorflow ONNX
·
MLOps
기본적인 사용법만을 작성한 글입니다. 모델 성능 확인을 위해 MNIST 데이터로 학습하는 과정을 추가했습니다. 참고한 사이트 tensorflow-onnx 목차 tf2onnx 설치 MNIST 모델 학습 및 저장 ONNX로 모델 변환 ONNX로 추론하기 추론 정확도 비교 추론 속도 비교 조금 더 복잡한 모델 결론 tf2onnx 설치 pip install -U tf2onnx # 현재 나의 버전은 1.10.0 # 알고있겠지만 -U는 --upgrade와 같다. MNIST 모델 학습 및 저장 import tensorflow as tf # 데이터 불러오기 (train_x, train_y), (test_x, test_y) = tf.keras.datasets.mnist.load_data() train_x, test_x ..
첫번째 오픈소스 기여 😆
·
ETC
개인 프로젝트인 Vinyl 하다가 pytube라는 라이브러리를 사용하려는데 Quick Start에서 HTTP 404 Error에러가 발생했다. Issue 탭에 가보니 발생한지 오래 되지 않은 치명적인 오류라는 것을 알 수 있었고 라이브러리 저자인 @tfdahlin가 문제를 해결하다 자러가며 다른 사용자에게 문제 해결을 부탁한 것을 보았다. 슈퍼맨도 아닌데 누가 도움을 요청하니까 꼭 나서서 해결하고 싶은 욕망이 생겼다. 열심히 디버깅해서 문제가 되는 부분을 찾았고 구글링해서 문제를 해결했다. 흥분되는 마음에 급하게 PR을 보냈다. 그런데도 Issue 탭에서는 문제에 대한 이야기가 오고갔다. 그래서 아래와 같이 Issue 탭에 글을 남겼다. 얼마 지나지 않아 고맙다는 댓글들이 올라오는 것을 보고 정말 큰 보..
라즈베리파이4 Pytorch 설치
·
라즈베리파이
라즈베리파이에 PyTorch를 설치하고 기본 MNIST 예제를 돌려보는 과정을 정리해보았다. 1. Update sudo apt-get update sudo apt-get dist-upgrade -y2. Install dependencies sudo apt install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools -y sudo apt-get install libavutil-dev libavcodec-dev libavformat-dev libswscale-dev -y3. Virtualenv Install sudo pip3 install virtualenv4. Make virtualenv and ..