fix type of args.scene

bertjiazheng · Oct 27, 2019 · 1cb4861 · 1cb4861
1 parent 06206a7
commit 1cb4861
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/visualize_3d.py b/visualize_3d.py
@@ -341,7 +341,7 @@ def parse_args():
     parser.add_argument("--path", required=True,
                         help="dataset path", metavar="DIR")
     parser.add_argument("--scene", required=True,
-                        help="scene id", type=str)
+                        help="scene id", type=int)
     parser.add_argument("--type", choices=("floorplan", "wireframe", "plane"),
                         default="plane", type=str)
     parser.add_argument("--color", choices=["normal", "manhattan"],