DIGITS (다른 레이블)에서 다른 데이터 세트로 새 사전 학습 모델을 사용하는 방법은 무엇입니까?

Question

VGG_ILSVRC_19_layers를 사전에 조상 된 모델이지만 다른 데이터 세트로 사용하고 싶습니다. 다른 라벨 파일이 필요합니까? 이 모델을 업로드하고 내 데이터 세트에 사용하려면 어떻게해야합니까?

VGG에 대한

16 층 나는

ERROR: Cannot copy param 0 weights from layer 'fc6'; shape mismatch. Source param shape is 1 1 4096 25088 (102760448); target param shape is 4096 32768 (134217728). To learn this layer's parameters from scratch rather than copying from a saved net, rename the layer.

방법 층을 수정할 수 있습니다

있어?

Answer 1

라벨이 모델이 아닌 DIGITS의 데이터 세트와 연결되어 있습니다. 미리 훈련 된 VGG 모델을 업로드 할 때 모델이 1000이 아닌 N 클래스에서 작동 할 수 있도록 마지막 내부 제품 계층의 이름을 바꿔야합니다 (this answer 참조).