Spaces:
Runtime error
Runtime error
joonhyun.jeong
commited on
Commit
·
7f3a3f1
1
Parent(s):
1c225cc
chore: add metadata wget
Browse files
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,
|