CurHarsh commited on
Commit
d69ba73
·
1 Parent(s): fb93448

Update cluster_visualize.py

Browse files
Files changed (1) hide show
  1. 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("./context_cluster/imagenet1k_id_to_label.txt", "r") as f:
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)