Skip to content

Commit

Permalink
imporved CLI for quick visu - polished details
Browse files Browse the repository at this point in the history
  • Loading branch information
JonasFrey96 committed Apr 1, 2024
1 parent 1bef9ea commit 2601fdd
Show file tree
Hide file tree
Showing 3 changed files with 368 additions and 300 deletions.
10 changes: 5 additions & 5 deletions quick_start.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,17 +56,17 @@ def parse_folders(args):
parser.add_argument(
"--input_image_folder",
default="demo_data",
help="If not gloabl will search for the folde name within the assests folder",
help="Gloabl path or folder name within the assests directory",
)
parser.add_argument(
"--output_folder_name",
default="demo_data",
help="If not global will create the folder within the results folder",
help="Gloabl path or folder name within the results directory",
)

# Fixed values
parser.add_argument("--network_input_image_height", type=int, default=224)
parser.add_argument("--network_input_image_width", type=int, default=224)
parser.add_argument("--network_input_image_height", type=int, default=224, help="Height of the input image")
parser.add_argument("--network_input_image_width", type=int, default=224, help="Width of the input image")
parser.add_argument(
"--segmentation_type",
default="stego",
Expand All @@ -89,7 +89,7 @@ def parse_folders(args):
parser.set_defaults(compute_confidence=True)

parser.add_argument(
"--prediction_per_pixel", action="store_true", help="Description of prediction per pixel argument"
"--prediction_per_pixel", action="store_true", help="Inference traversability per-pixel or per-segment"
)
parser.add_argument("--no-prediction_per_pixel", dest="prediction_per_pixel", action="store_false")
parser.set_defaults(prediction_per_pixel=True)
Expand Down
2 changes: 2 additions & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,8 @@
"liegroups@git+https://github.com/mmattamala/liegroups",
"wget",
"rospkg",
"wandb",
"opencv-python==4.2.0.34",
]
setup(
name="wild_visual_navigation",
Expand Down
Loading

0 comments on commit 2601fdd

Please sign in to comment.