Create fast_tiny_ic17mlt_640.json

fast_tiny_ic17mlt_640.json

@@ -0,0 +1,65 @@
+{
+    "model": {
+        "type": "FAST",
+        "backbone": {
+            "type": "fast_backbone",
+            "config": "config/fast/nas-configs/fast_tiny.config"
+        },
+        "neck": {
+            "type": "fast_neck",
+            "config": "config/fast/nas-configs/fast_tiny.config"
+        },
+        "detection_head": {
+            "type": "fast_head",
+            "config": "config/fast/nas-configs/fast_tiny.config",
+            "pooling_size": 9,
+            "loss_text": {
+                "type": "DiceLoss",
+                "loss_weight": 0.5
+            },
+            "loss_kernel": {
+                "type": "DiceLoss",
+                "loss_weight": 1.0
+            },
+            "loss_emb": {
+                "type": "EmbLoss_v1",
+                "feature_dim": 4,
+                "loss_weight": 0.25
+            }
+        }
+    },
+    "repeat_times": 10,
+    "data": {
+        "batch_size": 16,
+        "train": {
+            "type": "FAST_IC17MLT",
+            "split": "train",
+            "is_transform": true,
+            "img_size": 640,
+            "short_size": 640,
+            "pooling_size": 9,
+            "read_type": "cv2",
+            "repeat_times": 10
+        },
+        "test": {
+            "type": "FAST_IC17MLT",
+            "split": "valid",
+            "short_size": 640,
+            "read_type": "cv2"
+        }
+    },
+    "train_cfg": {
+        "lr": 0.001,
+        "schedule": "polylr",
+        "epoch": 30,
+        "optimizer": "Adam",
+        "save_interval": 1,
+        "pretrain": "/Users/eaxxkra/Downloads/fast_tiny_ic17mlt_640.pth"
+    },
+    "test_cfg": {
+        "result_path": "outputs/submit_ctw/",
+        "min_area": 250,
+        "min_score": 0.88,
+        "bbox_type": "rect"
+    }
+}