Exemplo n.º 1
0
def parse_args():
    parser = cli.ArgsParser()
    parser.add_argument(
        "--weight_type",
        default='pretrain',
        type=str,
        help=
        "Loading Checkpoints only support 'pretrain', 'finetune', 'resume'.")
    parser.add_argument("--eval",
                        action='store_true',
                        default=False,
                        help="Whether to perform evaluation in train")
    parser.add_argument("--slim_config",
                        default=None,
                        type=str,
                        help="Configuration file of slim method.")
    parser.add_argument("--enable_ce",
                        type=bool,
                        default=False,
                        help="If set True, enable continuous evaluation job."
                        "This flag is only used for internal test.")
    parser.add_argument("--fp16",
                        action='store_true',
                        default=False,
                        help="Enable mixed precision training.")
    parser.add_argument("--fleet",
                        action='store_true',
                        default=False,
                        help="Use fleet or not")
    args = parser.parse_args()
    return args
Exemplo n.º 2
0
def parse_args():
    parser = cli.ArgsParser()
    parser.add_argument("--eval",
                        action='store_true',
                        default=False,
                        help="Whether to perform evaluation in train")
    parser.add_argument("-r",
                        "--resume",
                        default=None,
                        help="weights path for resume")
    parser.add_argument("--slim_config",
                        default=None,
                        type=str,
                        help="Configuration file of slim method.")
    parser.add_argument("--enable_ce",
                        type=bool,
                        default=False,
                        help="If set True, enable continuous evaluation job."
                        "This flag is only used for internal test.")
    parser.add_argument("--amp",
                        action='store_true',
                        default=False,
                        help="Enable auto mixed precision training.")
    parser.add_argument("--fleet",
                        action='store_true',
                        default=False,
                        help="Use fleet or not")
    parser.add_argument("--use_vdl",
                        type=bool,
                        default=False,
                        help="whether to record the data to VisualDL.")
    parser.add_argument('--vdl_log_dir',
                        type=str,
                        default="vdl_log_dir/scalar",
                        help='VisualDL logging directory for scalar.')
    parser.add_argument('--save_prediction_only',
                        action='store_true',
                        default=False,
                        help='Whether to save the evaluation results only')
    parser.add_argument('--profiler_options',
                        type=str,
                        default=None,
                        help="The option of profiler, which should be in "
                        "format \"key1=value1;key2=value2;key3=value3\"."
                        "please see ppdet/utils/profiler.py for detail.")
    parser.add_argument('--save_proposals',
                        action='store_true',
                        default=False,
                        help='Whether to save the train proposals')
    parser.add_argument('--proposals_path',
                        type=str,
                        default="sniper/proposals.json",
                        help='Train proposals directory')

    args = parser.parse_args()
    return args
Exemplo n.º 3
0
def parse_args():
    parser = cli.ArgsParser()
    parser.add_argument(
        "-c",
        "--config",
        type=str,
        default='configs/yolov3/yolov3_mobilenet_v1_roadsign.yml',
        help="configuration file to use")
    parser.add_argument("--eval",
                        action='store_true',
                        default=False,
                        help="Whether to perform evaluation in train")
    parser.add_argument("-r",
                        "--resume",
                        default=None,
                        help="weights path for resume")
    parser.add_argument("--slim_config",
                        default=None,
                        type=str,
                        help="Configuration file of slim method.")
    parser.add_argument("--enable_ce",
                        type=bool,
                        default=False,
                        help="If set True, enable continuous evaluation job."
                        "This flag is only used for "
                        "internal test.")
    parser.add_argument("--fp16",
                        action='store_true',
                        default=False,
                        help="Enable mixed precision training.")
    parser.add_argument("--fleet",
                        action='store_true',
                        default=False,
                        help="Use fleet or not")
    parser.add_argument("--use_vdl",
                        type=bool,
                        default=False,
                        help="whether to record the data to VisualDL.")
    parser.add_argument('--vdl_log_dir',
                        type=str,
                        default="vdl_log_dir/scalar",
                        help='VisualDL logging directory for scalar.')
    parser.add_argument('--save_prediction_only',
                        action='store_true',
                        default=False,
                        help='Whether to save the evaluation results only')
    args = parser.parse_args()
    return args
Exemplo n.º 4
0
def parse_args():
    parser = cli.ArgsParser()
    parser.add_argument(
        "--weight_type",
        default='pretrain',
        type=str,
        help="Loading Checkpoints only support 'pretrain', 'finetune', 'resume'."
    )
    parser.add_argument(
        "--fp16",
        action='store_true',
        default=False,
        help="Enable mixed precision training.")
    parser.add_argument(
        "--loss_scale",
        default=8.,
        type=float,
        help="Mixed precision training loss scale.")
    parser.add_argument(
        "--eval",
        action='store_true',
        default=False,
        help="Whether to perform evaluation in train")
    parser.add_argument(
        "--output_eval",
        default=None,
        type=str,
        help="Evaluation directory, default is current directory.")
    parser.add_argument(
        "--enable_ce",
        type=bool,
        default=False,
        help="If set True, enable continuous evaluation job."
        "This flag is only used for internal test.")
    parser.add_argument(
        "--use_gpu", action='store_true', default=False, help="data parallel")
    args = parser.parse_args()
    return args