keras

    Lego Image를 분류하는 model 만들어보기

    학교 졸업작품으로 컴퓨터비전 기반 Lego sorting machine을 만들게 되었다. 컨베이어 벨트위에 있는 Lego를 Detect하는 부분은 OpenCV로 개발하였고 OpenCV가 전달해준 Lego Image를 어떤 블럭인지 Classification하는 모델을 만들어보았다. 모델 개요 88x88 그레이스케일 입력을 받아서 Prediction 결과로 총 11개의 Lego block중 1개를 출력한다. loss 함수 : categorical_crossentropy optimizer : Adam 사용된 학습관련 스킬 - 배치 정규화 - 가중치 초기화(he_normal) - 1x1 convolution - Dropout Layer 모델 소스코드 In [1]: import numpy as np import..

    Keras Model을 TensorRT로 변환하기

    시작 Jetson nano에서 Model 동작 속도를 높이기 위해 TensorRT로 최적화 하는 방법을 선택하였다. 로컬에 TensorRT 환경 구축을 하지 않아도 Google Colab을 통해 TensorRT를 사용할수 있다. Keras로 만든 Model File(.h5)을 Colab에 Upload하고 TensorRT를 설치후 변환해서 로컬에 다운로드 받으면 끝! 결과 Model : 간단한 Cnn Image Classification Model Device : Jetson nano Maxn Mode 변환전 : 4 프레임 변환 후(FP16) : 15 프레임 TensorRT로 변환 하는법 다음 링크를 참고해서 진행하면 된다. https://colab.research.google.com/drive/1U2MV..