IBM_AI/vi-opton/opton_od_train/template/config.py

from vi_core.od_config import od_config

class config(od_config):

    def get_train_classes(self):
        return ('__background__', # always index 0
            'segband'
            )