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))
