Проблемы с импортом .onnx файла обученной сети (MATLAB) после экспорта - PullRequest
0 голосов
/ 24 марта 2020

Я пытаюсь импортировать сеть, которую я сам обучил, используя этот код. Этот код запускается дважды. Сбой при второй попытке:

classes = ["Parasitized" "Uninfected"];

malaria_alexnet = 'C:\<directory>\malaria_alexnet_step2_accuracy_0.95952.onnx';
malaria_alexnet = importONNXNetwork(malaria_alexnet,'OutputLayerType','classification','Classes',classes);

lgraph = layerGraph(malaria_alexnet.Layers);
malaria_alexnet = trainNetwork(augimdsTrain,lgraph,options);

Причина, по которой я использую lgraph = layerGraph (malaria_ale xnet .Layers); , заключается в том, что я не хочу сбрасывать свои веса .

Однако, когда я импортирую эту сеть во второй раз, я получаю эту ошибку:

Error using nnet.internal.cnn.util.validateLayersForLayerGraph>iAssertUniqueAndNonEmptyLayerNames (line 56)
Layer names in layer array must be different from the names of layers in layer graph.

Error in nnet.internal.cnn.util.validateLayersForLayerGraph (line 33)
iAssertUniqueAndNonEmptyLayerNames(larray, existingLayers);

Error in nnet.cnn.LayerGraph>iValidateLayers (line 551)
larray = nnet.internal.cnn.util.validateLayersForLayerGraph(larray, existingLayers);

Error in nnet.cnn.LayerGraph/addLayers (line 190)
            larray = iValidateLayers(larray, existingLayers);

Error in nnet.internal.cnn.onnx.translateONNX>makeConnections (line 301)
    NNTLayerGraph = addLayers(NNTLayerGraph, NNTLayers{L});

Error in nnet.internal.cnn.onnx.translateONNX (line 131)
    NNTLayerGraphOrArray = makeConnections(NNTLayers, thisGraph, initializerNames);

Error in nnet.internal.cnn.onnx.importONNXNetwork (line 8)
LayersOrGraph   = nnet.internal.cnn.onnx.translateONNX(modelProto, OutputLayerType, UserImageInputSize, true);

Error in importONNXNetwork (line 52)
Network = nnet.internal.cnn.onnx.importONNXNetwork(modelfile, varargin{:});

Error in sgd_with_restarts_imported_network (line 17)
malaria_alexnet = importONNXNetwork(malaria_alexnet,'OutputLayerType','classification','Classes',classes);

Как мне импортировать мой файл .onnx или изменить имена слоев на что-то отличное от того, что находится в слое графа?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...