예제 #1
0
def arg_parse_params():
    """ parse the input parameters
    :return dict: {str: any}
    """
    # SEE: https://docs.python.org/3/library/argparse.html
    parser = argparse.ArgumentParser()
    parser.add_argument('--padding', type=float, required=False, default=0.1,
                        help='padding around the object in image percents')
    args = args_expand_parse_images(parser, NB_THREADS, overwrite=False)
    logging.info('ARGUMENTS: \n%r' % args)
    return args
예제 #2
0
def arg_parse_params():
    """ parse the input parameters
    :return dict: parameters
    """
    # SEE: https://docs.python.org/3/library/argparse.html
    parser = argparse.ArgumentParser()
    parser.add_argument('--dimension', type=int, required=False, choices=[0, 1],
                        help='cutting dimension', default=CUT_DIMENSION)
    args = args_expand_parse_images(parser, NB_WORKERS)
    logging.info('ARGUMENTS: \n%r' % args)
    return args
예제 #3
0
def arg_parse_params():
    """ parse the input parameters
    :return dict: parameters
    """
    # SEE: https://docs.python.org/3/library/argparse.html
    parser = argparse.ArgumentParser()
    parser.add_argument('-l',
                        '--level',
                        type=int,
                        required=False,
                        help='list of output scales',
                        default=DEFAULT_LEVEL)
    args = args_expand_parse_images(parser, NB_WORKERS)
    logging.info('ARGUMENTS: \n%r' % args)
    return args
예제 #4
0
def arg_parse_params():
    """ parse the input parameters
    :return dict: {str: any}
    """
    # SEE: https://docs.python.org/3/library/argparse.html
    parser = argparse.ArgumentParser()
    parser.add_argument('--scales',
                        type=int,
                        required=False,
                        nargs='+',
                        help='list of output scales',
                        default=DEFAULT_SCALES)
    parser.add_argument('-ext',
                        '--image_extension',
                        type=str,
                        required=False,
                        help='output image extension',
                        default=IMAGE_EXTENSION)
    args = args_expand_parse_images(parser, NB_THREADS)
    if not is_iterable(args['scales']):
        args['scales'] = [args['scales']]
    logging.info('ARGUMENTS: \n%r' % args)
    return args
예제 #5
0
 def test_argparse_images(self):
     with patch('argparse._sys.argv', ['script.py', '-i', 'an_image.png']):
         args = args_expand_parse_images(argparse.ArgumentParser())
         self.assertIsInstance(args, dict)