Ejemplo n.º 1
0
    def get_test_parser(self, config_path):
        parser = argparse.ArgumentParser(
            parents=[test_args_parser(config_path)], add_help=False)
        parser.add_argument('--wider-dir',
                            help='Location of WiderFace dataset.',
                            default='data/wider_face')

        return parser
Ejemplo n.º 2
0
def parse_args(template_filename):
    """ Parses input args. """

    parser = argparse.ArgumentParser(
        parents=[test_args_parser(template_filename)], add_help=False)
    parser.add_argument('--wider-dir',
                        help='Location of WiderFace dataset.',
                        default='data/wider_face')
    return parser.parse_args()
Ejemplo n.º 3
0
    def get_test_parser(self, config_path):
        parser = argparse.ArgumentParser(
            parents=[test_args_parser(config_path)], add_help=False)
        parser.add_argument(
            '--classes',
            required=True,
            help='Comma-separated list of classes (e.g. "cat,dog,mouse").')

        return parser
Ejemplo n.º 4
0
def parse_args(template_filename):
    """ Parses input args. """

    parser = argparse.ArgumentParser(
        parents=[test_args_parser(template_filename)], add_help=False)
    parser.add_argument(
        '--classes',
        required=True,
        help='Comma-separated list of classes (e.g. "cat,dog,mouse").')
    return parser.parse_args()
Ejemplo n.º 5
0
 def get_test_parser(self, config_path):
     return test_args_parser(config_path)
Ejemplo n.º 6
0
# Copyright (C) 2020 Intel Corporation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions
# and limitations under the License.

import os

from ote import MODEL_TEMPLATE_FILENAME
from ote.api import test_args_parser
from ote.args_conversion import convert_test_args
from ote.evaluation.common import evaluate

args = vars(test_args_parser(MODEL_TEMPLATE_FILENAME).parse_args())
ote_args = convert_test_args(os.path.dirname(MODEL_TEMPLATE_FILENAME), args)
evaluate(**ote_args)
 def get_test_parser(self, config_path):
     parser = argparse.ArgumentParser(
         parents=[test_args_parser(config_path)], add_help=False)
     parser = self._add_aux_weights_load_arg(parser)
     parser = self._add_classes_arg(parser)
     return parser
Ejemplo n.º 8
0
 def get_test_parser(self, config_path):
     parser = argparse.ArgumentParser(parents=[test_args_parser(config_path)], add_help=False)
     return self._add_classes_arg(parser)