Add arguments.
(parser)
| 31 | |
| 32 | |
| 33 | def add_parser(parser): |
| 34 | """Add arguments.""" |
| 35 | parser.add_argument( |
| 36 | '--base-json', |
| 37 | type=str, |
| 38 | default=None, |
| 39 | help='json config file for split images') |
| 40 | parser.add_argument( |
| 41 | '--nproc', type=int, default=10, help='the procession number') |
| 42 | |
| 43 | # argument for loading data |
| 44 | parser.add_argument( |
| 45 | '--img-ann-dirs', |
| 46 | nargs='+', |
| 47 | type=str, |
| 48 | default=None, |
| 49 | help='images annotation dirs, must give a value') |
| 50 | |
| 51 | # argument for splitting image |
| 52 | parser.add_argument( |
| 53 | '--sizes', |
| 54 | nargs='+', |
| 55 | type=int, |
| 56 | default=[1024], |
| 57 | help='the sizes of sliding windows') |
| 58 | parser.add_argument( |
| 59 | '--gaps', |
| 60 | nargs='+', |
| 61 | type=int, |
| 62 | default=[512], |
| 63 | help='the steps of sliding widnows') |
| 64 | parser.add_argument( |
| 65 | '--rates', |
| 66 | nargs='+', |
| 67 | type=float, |
| 68 | default=[1.], |
| 69 | help='same as DOTA devkit rate, but only change windows size') |
| 70 | parser.add_argument( |
| 71 | '--img-rate-thr', |
| 72 | type=float, |
| 73 | default=0.6, |
| 74 | help='the minimal rate of image in window and window') |
| 75 | parser.add_argument( |
| 76 | '--iof-thr', |
| 77 | type=float, |
| 78 | default=0.7, |
| 79 | help='the minimal iof between a object and a window') |
| 80 | parser.add_argument( |
| 81 | '--no-padding', |
| 82 | action='store_true', |
| 83 | help='not padding patches in regular size') |
| 84 | parser.add_argument( |
| 85 | '--padding-value', |
| 86 | nargs='+', |
| 87 | type=int, |
| 88 | default=[0], |
| 89 | help='padding value, 1 or channel number') |
| 90 |