Как использовать новую модель предварительной обработки с различным набором данных в 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, вам, вероятно, потребуется переименовать свой последний внутренний слой продукта (см. this answer), чтобы ваша модель работала над N классами вместо 1000.