Spaces:
Running
Running
File size: 2,497 Bytes
9860a06 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
"""
Copyright (c) 2024-present Naver Cloud Corp.
This source code is licensed under the license found in the
LICENSE file in the root directory of this source tree.
"""
import os
import argparse
from config.config import config_
def str2bool(v):
if isinstance(v, bool):
return v
if v.lower() in ("yes", "true", "t", "y", "1"):
return True
elif v.lower() in ("no", "false", "f", "n", "0"):
return False
else:
raise argparse.ArgumentTypeError("Boolean value expected.")
def get_parser(verbose=False):
p = argparse.ArgumentParser("argparser", add_help=False)
p.add_argument(
"--data-root", type=str, default=config_.data_root, help="data root directory"
)
p.add_argument(
"--local_rank", type=int, default=int(os.getenv("LOCAL_RANK", "0"))
)
p.add_argument(
"--amp", type=str2bool, default=True
)
p.add_argument(
"--ddp", action="store_true"
)
p.add_argument(
"--random-seed", type=int, default=config_.random_seed
)
# network config
p.add_argument(
"--network-encoder",
type=str,
default=config_.network.encoder,
choices=["vit_b", "vit_l"],
)
p.add_argument(
"--network-decoder",
type=str,
default=config_.network.decoder,
choices=["zim", "sam"],
)
p.add_argument(
"--network-encode-kernel",
type=int,
default=config_.network.encode_kernel,
)
# evaluation config
p.add_argument(
"--eval-workers", type=int, default=config_.eval.workers,
)
p.add_argument(
"--eval-image-size", type=int, default=config_.eval.image_size,
)
p.add_argument(
"--eval-prompt-type", type=str, default=config_.eval.prompt_type,
)
p.add_argument(
"--eval-model-list", type=str, default=config_.eval.model_list,
)
p.add_argument(
"--eval-zim-weights",
type=str,
default=config_.eval.zim_weights,
)
p.add_argument(
"--eval-sam-weights",
type=str,
default=config_.eval.sam_weights,
)
# dataset config
p.add_argument(
"--dataset-valset", type=str, default=config_.dataset.valset,
)
p.add_argument(
"--dataset-data-type", type=str, default=config_.dataset.data_type,
)
p.add_argument(
"--dataset-data-list-txt", type=str, default=config_.dataset.data_list_txt,
)
return p
|