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..