add metadata

Python-Repository-Hub · Mar 25, 2022 · 90890d7 · 90890d7
1 parent e9672a1
commit 90890d7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -11,7 +11,7 @@ inference_test_output
 
 
 *.png
-*.json
+# *.json
 *.diff
 
 # compilation and distribution
@@ -53,6 +53,7 @@ _darcs
 /datasets/*
 !/datasets/*.*
 !/datasets/README.md
+!/datasets/metadata/
 /projects/*/datasets
 /models
 /snippet
diff --git a/datasets/metadata/lvis_v1_train_cat_info.json b/datasets/metadata/lvis_v1_train_cat_info.json
diff --git a/gtr/config.py b/gtr/config.py
@@ -7,6 +7,8 @@ def add_gtr_config(cfg):
     _C.MODEL.ROI_BOX_HEAD.USE_SIGMOID_CE = False # classification loss
     _C.MODEL.ROI_BOX_HEAD.PRIOR_PROB = 0.01
     _C.MODEL.ROI_BOX_HEAD.DELAY_CLS = False # classification after tracking
+    _C.MODEL.ROI_BOX_HEAD.CAT_FREQ_PATH = \
+        'datasets/metadata/lvis_v1_train_cat_info.json' # LVIS metadata
 
     # association head
     _C.MODEL.ASSO_ON = False