Added shorter pascal configs.

lxk-yb · Jun 16, 2019 · 0d1f5b7 · 0d1f5b7
1 parent 3e2327b
commit 0d1f5b7
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/data/config.py b/data/config.py
@@ -703,6 +703,24 @@ def print(self):
     'num_classes': len(pascal2007_dataset.class_names) + 1,
 })
 
+yolact_base_pascal2_config = yolact_base_pascal_config.copy({
+    'name': 'yolact_base_pascal2',
+
+    'lr_steps': (70000, 150000, 175000, 187500),
+    'max_iter': 200000,
+})
+
+yolact_resnet50_pascal2_config = yolact_resnet50_config.copy({
+    'name': 'yolact_resnet50_pascal2',
+
+    # Dataset stuff
+    'dataset': pascal2007_dataset,
+    'num_classes': len(pascal2007_dataset.class_names) + 1,
+
+    'lr_steps': (70000, 150000, 175000, 187500),
+    'max_iter': 200000,
+})
+