Chan
commited on
Commit
•
a05e7bc
1
Parent(s):
2536e13
fix onnx path
Browse files- 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='
|
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')
|