参考链接：

import tensorflow as tfimport deepdish as ddimport argparseimport osimport numpy as npdef tr(v):    # tensorflow weights to pytorch weights    if v.ndim == 4:        return np.ascontiguousarray(v.transpose(3,2,0,1))    elif v.ndim == 2:        return np.ascontiguousarray(v.transpose())    return vdef read_ckpt(ckpt):    # https://github.com/tensorflow/tensorflow/issues/1823    reader = tf.train.NewCheckpointReader(ckpt)    weights = {n: reader.get_tensor(n) for (n, _) in reader.get_variable_to_shape_map().iteritems()}    pyweights = {k: tr(v) for (k, v) in weights.items()}    return pyweightsif __name__ == '__main__':    parser = argparse.ArgumentParser(description="Converts ckpt weights to deepdish hdf5")    parser.add_argument("infile", type=str,                        help="Path to the ckpt.")    parser.add_argument("outfile", type=str, nargs='?', default='',                        help="Output file (inferred if missing).")    args = parser.parse_args()    if args.outfile == '':        args.outfile = os.path.splitext(args.infile)[0] + '.h5'    outdir = os.path.dirname(args.outfile)    if not os.path.exists(outdir):        os.makedirs(outdir)    weights = read_ckpt(args.infile)    dd.io.save(args.outfile, weights)    weights2 = dd.io.load(args.outfile)