()
| 392 | |
| 393 | |
| 394 | def parse_args(): |
| 395 | parser = argparse.ArgumentParser(description='Train RetinaFace') |
| 396 | # general |
| 397 | parser.add_argument('--network', |
| 398 | help='network name', |
| 399 | default=default.network, |
| 400 | type=str) |
| 401 | parser.add_argument('--dataset', |
| 402 | help='dataset name', |
| 403 | default=default.dataset, |
| 404 | type=str) |
| 405 | args, rest = parser.parse_known_args() |
| 406 | generate_config(args.network, args.dataset) |
| 407 | parser.add_argument('--image_set', |
| 408 | help='image_set name', |
| 409 | default=default.image_set, |
| 410 | type=str) |
| 411 | parser.add_argument('--root_path', |
| 412 | help='output data folder', |
| 413 | default=default.root_path, |
| 414 | type=str) |
| 415 | parser.add_argument('--dataset_path', |
| 416 | help='dataset path', |
| 417 | default=default.dataset_path, |
| 418 | type=str) |
| 419 | # training |
| 420 | parser.add_argument('--frequent', |
| 421 | help='frequency of logging', |
| 422 | default=default.frequent, |
| 423 | type=int) |
| 424 | parser.add_argument('--kvstore', |
| 425 | help='the kv-store type', |
| 426 | default=default.kvstore, |
| 427 | type=str) |
| 428 | parser.add_argument('--work_load_list', |
| 429 | help='work load for different devices', |
| 430 | default=None, |
| 431 | type=list) |
| 432 | parser.add_argument('--no_flip', |
| 433 | help='disable flip images', |
| 434 | action='store_true') |
| 435 | parser.add_argument('--no_shuffle', |
| 436 | help='disable random shuffle', |
| 437 | action='store_true') |
| 438 | # e2e |
| 439 | #parser.add_argument('--gpus', help='GPU device to train with', default='0,1,2,3', type=str) |
| 440 | parser.add_argument('--pretrained', |
| 441 | help='pretrained model prefix', |
| 442 | default=default.pretrained, |
| 443 | type=str) |
| 444 | parser.add_argument('--pretrained_epoch', |
| 445 | help='pretrained model epoch', |
| 446 | default=default.pretrained_epoch, |
| 447 | type=int) |
| 448 | parser.add_argument('--prefix', |
| 449 | help='new model prefix', |
| 450 | default=default.prefix, |
| 451 | type=str) |
no test coverage detected