Update cluster_visualize.py
Browse files- cluster_visualize.py +1 -1
cluster_visualize.py
CHANGED
@@ -19,7 +19,7 @@ from timm.models import load_checkpoint
|
|
19 |
from torchvision.utils import draw_segmentation_masks
|
20 |
|
21 |
object_categories = []
|
22 |
-
with open("
|
23 |
for line in f:
|
24 |
_, val = line.strip().split(":")
|
25 |
object_categories.append(val)
|
|
|
19 |
from torchvision.utils import draw_segmentation_masks
|
20 |
|
21 |
object_categories = []
|
22 |
+
with open("imagenet1k_id_to_label.txt", "r") as f:
|
23 |
for line in f:
|
24 |
_, val = line.strip().split(":")
|
25 |
object_categories.append(val)
|