딥러닝/컴퓨터비전

[컴퓨터비전3]CNN 연산 시 사용하는 메모리, 파라미터 계산

데이터요리사 2021. 12. 13. 12:24

ci-1 = 이전 층의 채널(필터) 수

ci = 현재 층의 채널(필터) 수

mi-1 = 이전 피처맵 사이즈

mi = 현재 피처맵 사이즈

k = 필터 사이즈

구분 유형 수식 설명
Storage 파라미터 수 k x k x ci-1 x ci = 가중치 수
 . 학습 이후에 메모리에 저장되는 값
뉴런 수 mi x mi x ci = 메모리 사용량
. 학습이 완료된 신경망을 사용하여 추론(reference)할 때 차지하는 메모리 사이즈
Computational  OPS
(FLOPS)
k x k x mi x mi x ci-1 x ci . 추론과정에서 연산 속도에 영향을 줌
. 학습과정에서는 그리 중요하지 않음

 

728x90