如何将.npy文件作为权重和偏差的恢复文件参数加载到tensorflow

时间:2020-02-29 13:00:04

标签: python tensorflow tracker

我在TensorFlow中使用MDNet跟踪器,但是无法恢复.npy格式的已保存参数文件

using Microsoft.OpenApi.Models;

但是,收到以下错误消息:

        # restore from model
        restore_path =         np.load(r"/home/ammar/Bureau/BSB/objectDetector/MDNet/TensorFlow/Tensorflow- 
        MDNet2/PyMDNet/models/init.npy", encoding='latin1', allow_pickle=True)
        print(restore_path.shape)
        saver.restore(sess, restore_path)

        # run mdnet
        mdnet_run(sess, model, train_data.data[seq].gts[0], train_data.data[seq].frames, config, display)


    def get_params():
        parser = argparse.ArgumentParser()
        parser.add_argument('--no_display', action='store_true', help='disable display')
        parser.add_argument('--dataset', choices=['otb', 'vot2013', 'vot2014', 'vot2015'],
                            help='choose pretrained dataset: [otb/vot2013/vot2014/vot2015]')
        parser.add_argument('--seq', default=None, help='specify the sequence name')
        # parser.add_argument('--load_path', default=None, help='initial model path')
        return parser.parse_args()


    if __name__ == '__main__':
        params = get_params()
        tracking(params.dataset, params.seq, display=(not params.no_display))

我正在使用Python版本= 3.6,TensorFlow = 1.10和Numpy = 1.18: 无法在python 3中加载.npy文件,或者在加载文件的方式上有问题吗?

0 个答案:

没有答案
相关问题