def __init__(self, name: str, label_path: pathlib.Path): self.name: str = name label_map = label_map_util.load_labelmap(str(label_path)) self.categories = label_map_util.convert_label_map_to_categories( label_map, max_num_classes=1000) self.category_index: CategoryIndex = ( label_map_util.create_category_index(self.categories))
def __init__(self, graph: tf.Graph, label_map: StringIntLabelMap): self.graph: tf.Graph = graph self.label_map: StringIntLabelMap = label_map self.categories = label_map_util.convert_label_map_to_categories( label_map, max_num_classes=1000) self.category_index: CategoryIndex = ( label_map_util.create_category_index(self.categories))