Как извлечь сетевые коэффициенты из тензорного потока.ckpt в.npy?
Я работаю с реализацией тензорного потока FasterRCNN, найденной ЗДЕСЬ. При определении функции train_net в файле train.py они определяют заставку с помощью следующей команды:
saver = tf.train.Saver(max_to_keep=100, write_version=saver_pb2.SaverDef.V1)
В результате в конце обучающего цикла в выходной папке создается несколько файлов.ckpt, каждый из которых содержит снимок коэффициентов модели для (по умолчанию) каждых 5000 шагов итерации обучения. Теперь, если вы попытаетесь запустить процесс тестирования (или если вы хотите продолжить обучение), у вас есть 2 варианта загрузки предварительно обученной модели. Либо вы загружаете один такой файл.ckpt, либо вы можете загрузить файл.npy, например, популярный VGG_imagenet.npy (два случая обрабатываются в функции 'load' в классе 'Network', определенном в файле 'network.py.
Есть ли способ преобразовать сетевые коэффициенты, сохраненные в файле.ckpt, в файл.npy со структурой VGG_imagenet.npy? Можно ли получить это, импортировав несколько строк кода сразу после загрузки информации.ckpt (в разделе инициализации переменных SolverWrapper>train_model() в файле train.py)?
Спасибо.