fix ShTech keyError

LiUzHiAn · Dec 20, 2021 · 0f34301 · 0f34301
1 parent e377d6e
commit 0f34301
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pre_process/extract_bboxes.py b/pre_process/extract_bboxes.py
@@ -13,7 +13,7 @@
 DATASET_CFGS = {
     "ped2": {"conf_thr": 0.5, "min_area": 10 * 10, "cover_thr": 0.6, "binary_thr": 18, "gauss_mask_size": 3, 'contour_min_area': 10 * 10},
     "avenue": {"conf_thr": 0.25, "min_area": 40 * 40, "cover_thr": 0.6, "binary_thr": 18, "gauss_mask_size": 5, 'contour_min_area': 40 * 40},
-    "our_shanghaitech": {"conf_thr": 0.5, "min_area": 8 * 8, "cover_thr": 0.65, "binary_thr": 15, "gauss_mask_size": 5, 'contour_min_area': 40 * 40}
+    "shanghaitech": {"conf_thr": 0.5, "min_area": 8 * 8, "cover_thr": 0.65, "binary_thr": 15, "gauss_mask_size": 5, 'contour_min_area': 40 * 40}
 }