У меня есть модель кафе, в которой есть crop слоев, поэтому преобразование в TensorFlow создает проблему.
Я успешно загрузил модель и удалил слои crop, и теперь я хотел бы сохранить соответствующие model.prototxt и model.caffemodel
Я нашел следующие вопросы о StackOverflow, но они касаются замены слоев, а не их окончательного удаления:
Удаление слоя и обновление модели кафе
Как изменить модель кофе Imag enet?
Когда я сохраняю модель, используя caffe.Net.save(), сохраняется только файл model.caffemodel, а не соответствующий .prototxt. Что делать?
Файлы моделей: https://github.com/Charrin/RetinaFace-Cpp/tree/master/convert_models/mnet
Код, используемый до сих пор -
import caffe
net = caffe.Net('mnet.prototxt', 'mnet.caffemodel' , caffe.TEST)
del net.layer_dict['crop1']
del net.layer_dict['crop0']
net.save('new_model.caffemodel')