argmax

21.07.27 23:15 작성 조회수 36

삭제된 글입니다

[파이토치] 실전 인공지능으로 이어지는 딥러닝 - 기초부터 논문 구현까지

이미지 분류 - 합성곱 신경망(CNN)

강의실 바로가기

답변 1

답변을 작성해보세요.

딥러닝호형

2021.07.28

안녕하세요!

여기서는 Flatten하고 관계는 없습니다.

outputs 크기가 (배치 사이즈)x(클래스 수) 입니다. 즉, 각 이미지 당 클래스 수 만큼의 크기의 벡터로 표현 됩니다. 따라서 예측값을 정할 때 각 벡터마다 가장 큰 값을 기준으로 예측값을 정하기 때문에 열 기준으로 큰 값을 산정합니다.

예를 들어 클래스가 3개짜리인 문제에 배치 크기가 2인 outputs= [[0.1, 0.4, 0.5],[0.5 ,0.3, 0.2]] 라고 하면 각 열 기준 첫번째 이미지의 예측값은 2이고 두번째 예측값은 0이 되는 것입니다 :)

Jaewoo Choi

2021.07.28

헉,,,감사합니다..그냥 넘어갔으면 평생을 모르고 넘어갈뻔했내여 :)