(self)
| 8 | |
| 9 | class opts(object): |
| 10 | def __init__(self): |
| 11 | self.parser = argparse.ArgumentParser() |
| 12 | # basic experiment setting |
| 13 | self.parser.add_argument('task', default='', |
| 14 | help='ctdet | ddd | multi_pose ' |
| 15 | '| tracking or combined with ,') |
| 16 | self.parser.add_argument('--dataset', default='coco', |
| 17 | help='see lib/dataset/dataset_facotry for ' + |
| 18 | 'available datasets') |
| 19 | self.parser.add_argument('--test_dataset', default='', |
| 20 | help='coco | kitti | coco_hp | pascal') |
| 21 | self.parser.add_argument('--exp_id', default='default') |
| 22 | self.parser.add_argument('--test', action='store_true') |
| 23 | self.parser.add_argument('--debug', type=int, default=0, |
| 24 | help='level of visualization.' |
| 25 | '1: only show the final detection results' |
| 26 | '2: show the network output features' |
| 27 | '3: use matplot to display' # useful when lunching training with ipython notebook |
| 28 | '4: save all visualizations to disk') |
| 29 | self.parser.add_argument('--no_pause', action='store_true') |
| 30 | self.parser.add_argument('--demo', default='', |
| 31 | help='path to image/ image folders/ video. ' |
| 32 | 'or "webcam"') |
| 33 | self.parser.add_argument('--load_model', default='', |
| 34 | help='path to pretrained model') |
| 35 | self.parser.add_argument('--resume', action='store_true', |
| 36 | help='resume an experiment. ' |
| 37 | 'Reloaded the optimizer parameter and ' |
| 38 | 'set load_model to model_last.pth ' |
| 39 | 'in the exp dir if load_model is empty.') |
| 40 | |
| 41 | # system |
| 42 | self.parser.add_argument('--gpus', default='0', |
| 43 | help='-1 for CPU, use comma for multiple gpus') |
| 44 | self.parser.add_argument('--num_workers', type=int, default=4, |
| 45 | help='dataloader threads. 0 for single-thread.') |
| 46 | self.parser.add_argument('--not_cuda_benchmark', action='store_true', |
| 47 | help='disable when the input size is not fixed.') |
| 48 | self.parser.add_argument('--seed', type=int, default=317, |
| 49 | help='random seed') # from CornerNet |
| 50 | self.parser.add_argument('--not_set_cuda_env', action='store_true', |
| 51 | help='used when training in slurm clusters.') |
| 52 | |
| 53 | # log |
| 54 | self.parser.add_argument('--print_iter', type=int, default=0, |
| 55 | help='disable progress bar and print to screen.') |
| 56 | self.parser.add_argument('--save_all', action='store_true', |
| 57 | help='save model to disk every 5 epochs.') |
| 58 | self.parser.add_argument('--vis_thresh', type=float, default=0.3, |
| 59 | help='visualization threshold.') |
| 60 | self.parser.add_argument('--debugger_theme', default='white', |
| 61 | choices=['white', 'black']) |
| 62 | self.parser.add_argument('--eval_val', action='store_true') |
| 63 | self.parser.add_argument('--save_imgs', default='', help='') |
| 64 | self.parser.add_argument('--save_img_suffix', default='', help='') |
| 65 | self.parser.add_argument('--skip_first', type=int, default=-1, help='') |
| 66 | self.parser.add_argument('--save_video', action='store_true') |
| 67 | self.parser.add_argument('--save_framerate', type=int, default=30) |
nothing calls this directly
no outgoing calls
no test coverage detected