update config.yaml

config.yaml

@@ -1,65 +1,41 @@
-__include__: [
-  '../../dataset/custom_detection.yml',
-  '../../runtime.yml',
-  '../include/dataloader.yml',
-  '../include/optimizer.yml',
-  '../include/dfine_hgnetv2.yml',
-]
+task: detection
 
-output_dir: ./output/dfine_hgnetv2_s_custom
+evaluator:
+  type: CocoEvaluator
+  iou_types: ['bbox', ]
 
+num_classes: 3 # your dataset classes
+remap_mscoco_category: False
 
-DFINE:
-  backbone: HGNetv2
-
-HGNetv2:
-  name: 'B0'
-  return_idx: [1, 2, 3]
-  freeze_at: -1
-  freeze_norm: False
-  use_lab: True
-
-DFINETransformer:
-  num_layers: 3  # 4 5 6
-  eval_idx: -1  # -2 -3 -4
-
-HybridEncoder:
-  in_channels: [256, 512, 1024]
-  hidden_dim: 256
-  depth_mult: 0.34
-  expansion: 0.5
-
-optimizer:
-  type: AdamW
-  params:
-    -
-      params: '^(?=.*backbone)(?!.*norm|bn).*$'
-      lr: 0.0002
-    -
-      params: '^(?=.*backbone)(?=.*norm|bn).*$'
-      lr: 0.0002
-      weight_decay: 0.
-    -
-      params: '^(?=.*(?:encoder|decoder))(?=.*(?:norm|bn|bias)).*$'
-      weight_decay: 0.
-
-  lr: 0.0004
-  betas: [0.9, 0.999]
-  weight_decay: 0.0001
-
-
-# Increase to search for the optimal ema
-epochs: 220
 train_dataloader:
-  total_batch_size: 64
+  type: DataLoader
   dataset:
+    type: CocoDetection
+    img_folder: /dataset/images/train
+    ann_file: /dataset/annotations/instances_train.json
+    return_masks: False
     transforms:
-      policy:
-        epoch: 200
+      type: Compose
+      ops: ~
+  shuffle: True
+  num_workers: 4
+  drop_last: True
   collate_fn:
-    stop_epoch: 200
-    ema_restart_decay: 0.9999
-    base_size_repeat: 20
+    type: BatchImageCollateFunction
+
 
 val_dataloader:
-  total_batch_size: 128
+  type: DataLoader
+  dataset:
+    type: CocoDetection
+    img_folder: /dataset/images/val
+    ann_file: /dataset/annotations/instances_val.json
+    return_masks: False
+    transforms:
+      type: Compose
+      ops: ~
+  shuffle: False
+  num_workers: 4
+  drop_last: False
+  collate_fn:
+    type: BatchImageCollateFunction