joonhyun.jeong commited on
Commit
7f3a3f1
·
1 Parent(s): 1c225cc

chore: add metadata wget

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -25,6 +25,14 @@ zs_weight_path = 'datasets/metadata/lvis_v1_clip_a+cname.npy'
25
  if not os.path.exists(zs_weight_path):
26
  wget.download("https://github.com/facebookresearch/Detic/raw/main/datasets/metadata/lvis_v1_clip_a+cname.npy", out=zs_weight_path)
27
 
 
 
 
 
 
 
 
 
28
  args = Namespace(
29
  base_cat_threshold=0.9,
30
  confidence_threshold=0.0,
 
25
  if not os.path.exists(zs_weight_path):
26
  wget.download("https://github.com/facebookresearch/Detic/raw/main/datasets/metadata/lvis_v1_clip_a+cname.npy", out=zs_weight_path)
27
 
28
+ base_cat_mask_path = "datasets/metadata/lvis_v1_base_cat_mask.npy"
29
+ if not os.path.exists(base_cat_mask_path):
30
+ wget.download("https://docs.google.com/uc?export=download&id=1CbSs5yeqMsWDkRSsIlB-ln_bXDv686rH", out=base_cat_mask_path)
31
+
32
+ lvis_train_cat_info_path = "datasets/metadata/lvis_v1_train_cat_info.json"
33
+ if not os.path.exists(lvis_train_cat_info_path):
34
+ wget.download("https://docs.google.com/uc?export=download&id=17WmkAJYBK4xT-YkiXLcwIWmtfulSUtmO", out=lvis_train_cat_info_path)
35
+
36
  args = Namespace(
37
  base_cat_threshold=0.9,
38
  confidence_threshold=0.0,