Skip to content

Commit

Permalink
better defaults
Browse files Browse the repository at this point in the history
  • Loading branch information
vict0rsch committed Jan 21, 2021
1 parent 4f22515 commit f497d34
Showing 1 changed file with 9 additions and 9 deletions.
18 changes: 9 additions & 9 deletions apply_events.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def parse_args():
"-b",
"--batch_size",
type=int,
default=8,
default=4,
help="Batch size to process input images to events",
)
parser.add_argument(
Expand All @@ -127,11 +127,10 @@ def parse_args():
+ " In particular it must contain opts.yaml and checkpoints/",
)
parser.add_argument(
"-t",
"--time",
"--no_time",
action="store_true",
default=False,
help="Binary flag to time operations or not. Defaults to False.",
help="Binary flag to prevent the timing of operations.",
)
parser.add_argument(
"-m",
Expand Down Expand Up @@ -175,11 +174,11 @@ def parse_args():
help="Do not check for existing outdir",
)
parser.add_argument(
"-c",
"--cloudy",
"--no_cloudy",
action="store_true",
default=False,
help="Use the cloudy intermediate image to create the flood image",
help="Prevent the use of the cloudy intermediate"
+ " image to create the flood image",
)
parser.add_argument(
"--keep_ratio_128",
Expand Down Expand Up @@ -210,6 +209,7 @@ def parse_args():

batch_size = args.batch_size
half = args.half
cloudy = not args.no_cloudy
images_paths = Path(args.images_paths).expanduser().resolve()
bin_value = args.flood_mask_binarization
outdir = (
Expand All @@ -218,7 +218,7 @@ def parse_args():
else None
)
resume_path = args.resume_path
time_inference = args.time
time_inference = not args.no_time
n_images = args.n_images
xla_purge_samples = args.xla_purge_samples
if args.keep_ratio_128:
Expand Down Expand Up @@ -351,7 +351,7 @@ def parse_args():
bin_value=bin_value,
half=half,
xla=XLA,
cloudy=args.cloudy,
cloudy=cloudy,
)

# store events to write after inference loop
Expand Down

0 comments on commit f497d34

Please sign in to comment.