Python save_graph 예제들

예제 #1

파일: convert_pkl_to_pb.py 프로젝트: sitingshen/Detectron-ZZNet

def _save_image_graphs(args, all_net, all_init_net):
    print('Saving model graph...')
    mutils.save_graph(
        all_net.Proto(), os.path.join(args.out_dir, "model_def.png"),
        op_only=False)
    print('Model def image saved to {}.'.format(args.out_dir))

예제 #2

파일: convert_pkl_to_pb.py 프로젝트: Jhird/Detectron

def _save_image_graphs(args, all_net, all_init_net):
    print('Saving model graph...')
    mutils.save_graph(
        all_net.Proto(), os.path.join(args.out_dir, "model_def.png"),
        op_only=False)
    print('Model def image saved to {}.'.format(args.out_dir))