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