from .registry import DATASETS
from .xml_style import XMLDataset


@DATASETS.register_module
class MeterDetection(XMLDataset):
    CLASSES = (
        '10002', '70012','70022',
        '80000',
        '80001',
        '80010',
        '80011',
        '80020',
        '80021',
        '80031',
        '80040',
        '80050',
        '80051',
        '80061',
        '80070',
        '80080',
        '80090',
        '80101',
        '80110',
        '80120',
        '81000',
        '81001',
        '81011',
        '81021',
        '81031',
        '81041',
        '81051',
        '81060',
        '81061',
        '81071',
        '81081',
        '81091',
        '82010',
        '82011',
        '82020',
        '82030',
        '82031',
        '82040',
        '82041',
        '82042',
        '82050',
        '82051',
        '82052',
        '82060',
        '82061',
        '82070',
        '82071',
        '82080',
        '82081',
        '82090',
        '82091',
        '82990',
        '83010',
        '83011',
        '83012',
        '83020',
        '83021',
        '83022',
        '83030',
        '83031',
        '83040',
        '83041',
        '83050',
        '83051',
        '83060',
        '83061',
        '83071',
        '83081',
        '83090',
        '83091',
    )
    
    def __init__(self, **kwargs):
        super(MeterDetection, self).__init__(**kwargs)

from .registry import DATASETS
from .xml_style import XMLDataset


@DATASETS.register_module
class MeterDetection(XMLDataset):
    CLASSES = (
        '10002', '70012','70022',
        '80000',
        '80001',
        '80010',
        '80011',
        '80020',
        '80021',
        '80031',
        '80040',
        '80050',
        '80051',
        '80061',
        '80070',
        '80080',
        '80090',
        '80101',
        '80110',
        '80120',
        '81000',
        '81001',
        '81011',
        '81021',
        '81031',
        '81041',
        '81051',
        '81060',
        '81061',
        '81071',
        '81081',
        '81091',
        '82010',
        '82011',
        '82020',
        '82030',
        '82031',
        '82040',
        '82041',
        '82042',
        '82050',
        '82051',
        '82052',
        '82060',
        '82061',
        '82070',
        '82071',
        '82080',
        '82081',
        '82090',
        '82091',
        '82990',
        '83010',
        '83011',
        '83012',
        '83020',
        '83021',
        '83022',
        '83030',
        '83031',
        '83040',
        '83041',
        '83050',
        '83051',
        '83060',
        '83061',
        '83071',
        '83081',
        '83090',
        '83091',
    )
    
    def __init__(self, **kwargs):
        super(MeterDetection, self).__init__(**kwargs)
