1

vgg-imagenet-f 네트워크를 재 분류하여 직접 이미지 비교가 아닌 분류 작업을 수행하고 싶습니다. 이는 직접 네트워크를 통해 수행 한 작업입니다. 그러나 다운로드 한 네트워크는 배포 네트워크이며 손실 계층은 포함되어 있지 않습니다. 이전에 분류 교육을 해본 적이 없으므로이 마지막 레이어를 디자인하는 방법에 대해 다소 혼란스러워합니다.matconvnet 분류 교육 마지막 레이어 (softmax)?

layer.name = 'loss' ; 
layer.type = 'custom' ; 
layer.forward = @forward ; 
layer.backward = @backward ; 
layer.class = [] ; 

하지만 난 내 @forward 및 @backward 기능을해야 모르겠어요 : 나는 이런 식으로 뭔가있을 것으로 기대. 그들은 softmax일까요? 약 10k 이미지, 해당 레이블 및 고유 번호가있는 ID 요소가 1 - 10k 인 imdb가 있습니다. matconvnet/matlab에서이 레이어를 구성하는 방법에 대한 도움이나 도움이되는 샘플에 대한 링크를 제공해 주셔서 감사합니다!

답변

0

임의의 숫자로 가중치를 초기화하는 대신 vgg를 '재 훈련'하려는 경우 다운로드 한 네트워크에서 훈련 된 파일러를 사용하여 분류 네트워크를 적용 할 수 있으므로 필터를 적절하게 조정하여 네트워크를 구현할 수 있습니다. 마지막 층은 softmaxloss 일 수 있습니다. http://www.vlfeat.org/matconvnet/mfiles/vl_nnsoftmaxloss/

관련 문제