Chan commited on
Commit
a05e7bc
1 Parent(s): 2536e13

fix onnx path

Browse files
Files changed (1) hide show
  1. infer_onnx.py +1 -1
infer_onnx.py CHANGED
@@ -19,7 +19,7 @@ from datasets.utils import colorize_mask, build_img
19
 
20
  if __name__ == "__main__":
21
  parser = argparse.ArgumentParser(description='SemanticFPN model')
22
- parser.add_argument('--onnx_path', type=str, default='FPN_int.onnx')
23
  parser.add_argument('--save_path', type=str, default='./data/demo_results/senmatic_results.png')
24
  parser.add_argument('--input_path', type=str, default='data/cityscapes/leftImg8bit/test/bonn/bonn_000000_000019_leftImg8bit.png')
25
  parser.add_argument('--ipu', action='store_true', help='use ipu')
 
19
 
20
  if __name__ == "__main__":
21
  parser = argparse.ArgumentParser(description='SemanticFPN model')
22
+ parser.add_argument('--onnx_path', type=str, default='FPN_int_NHWC.onnx')
23
  parser.add_argument('--save_path', type=str, default='./data/demo_results/senmatic_results.png')
24
  parser.add_argument('--input_path', type=str, default='data/cityscapes/leftImg8bit/test/bonn/bonn_000000_000019_leftImg8bit.png')
25
  parser.add_argument('--ipu', action='store_true', help='use ipu')