hewumars
10/9/2019 - 2:14 AM

pb转uff

import uff
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-m', action='store', dest="pb_path",default="/home/mars/hewu/Model/Cloud-Net.pb",help="pb path")
parser.add_argument('-on', action='store', dest="output_name",help="model output name")
parser.add_argument('-o', action='store', dest="uff_path",default="/home/mars/hewu/Model/Cloud-Net.uff",help="uff path")
args = parser.parse_args()
output_name=[args.output_name]
kwargs={'output_filename':args.uff_path}
uff.from_tensorflow_frozen_model(args.pb_path,output_name,**kwargs)