ai-training/AnnotationClass.py

import json


class AnnotationClass:
    def __init__(self, id, name, color):
        self.id = id
        self.name = name
        self.color = color

    @staticmethod
    def read_json():
        with open('classes.json', 'r', encoding='utf-8') as f:
            j = json.loads(f.read())
            return {cl['Id']: AnnotationClass(id=cl['Id'], name=cl['Name'], color=cl['Color']) for cl in j}

    @property
    def color_tuple(self):
        color = self.color[3:]
        lv = len(color)
        xx = range(0, lv, lv // 3)
        return tuple(int(color[i:i + lv // 3], 16) for i in xx)