from vi_core.od_config import od_config

class config(od_config):

    def get_train_classes(self):
        return ('__background__', # always index 0
            '10',
            '20',
            '50',
            '100',
            '200'
            )

    def get_model_confidence(self):
        train_classes = self.get_train_classes()
        d = {}
        for i in range(1, len(train_classes)):
            d[train_classes[i]] = 0
        return d