detector module

Created by: stanislavBoltenov

Изменения в cli_main.py

Изменения в функции cli_argument_parser()

    parser.add_argument('-m', '--model',
                        help='Model (\'MobileNet\')',
                        type=str,
                        dest='model',
                        choices=['MobileNet', 'YOLOv3-tiny'],
                        required=True,
                        default='MobileNet')
    parser.add_argument('-mp', '--model_path',
                        help='Path to a file of model',
                        type=str,
                        dest='model_path',
                        required=True)

Изменения в методе main()

     detector = Detector.create( args.model_path, args.model )

Найдена и исправлена ошибка в классе GroundtruthReader

При чтении файла с разметкой возникало исключение из-за некорректного вычисления количества элементов в строке.

    row[0] = row[0].split(' ')
    if len(row[0]) != 6: