Parse input arguments
()
| 11 | |
| 12 | |
| 13 | def parse_args(): |
| 14 | """Parse input arguments |
| 15 | """ |
| 16 | |
| 17 | parser = ArgumentParser(description=__doc__, |
| 18 | formatter_class=ArgumentDefaultsHelpFormatter) |
| 19 | |
| 20 | parser.add_argument('input_net_proto_file', |
| 21 | help='Input network prototxt file') |
| 22 | parser.add_argument('output_image_file', |
| 23 | help='Output image file') |
| 24 | parser.add_argument('--rankdir', |
| 25 | help=('One of TB (top-bottom, i.e., vertical), ' |
| 26 | 'RL (right-left, i.e., horizontal), or another ' |
| 27 | 'valid dot option; see ' |
| 28 | 'http://www.graphviz.org/doc/info/' |
| 29 | 'attrs.html#k:rankdir'), |
| 30 | default='LR') |
| 31 | parser.add_argument('--phase', |
| 32 | help=('Which network phase to draw: can be TRAIN, ' |
| 33 | 'TEST, or ALL. If ALL, then all layers are drawn ' |
| 34 | 'regardless of phase.'), |
| 35 | default="ALL") |
| 36 | |
| 37 | args = parser.parse_args() |
| 38 | return args |
| 39 | |
| 40 | |
| 41 | def main(): |