Update configs/dataset/custom_detection.yml

configs/dataset/custom_detection.yml

@@ -1,41 +1,41 @@
-task: detection
-
-evaluator:
-  type: CocoEvaluator
-  iou_types: ['bbox', ]
-
-num_classes: 777 # your dataset classes
-remap_mscoco_category: False
-
-train_dataloader:
-  type: DataLoader
-  dataset:
-    type: CocoDetection
-    img_folder: /data/yourdataset/train
-    ann_file: /data/yourdataset/train/train.json
-    return_masks: False
-    transforms:
-      type: Compose
-      ops: ~
-  shuffle: True
-  num_workers: 4
-  drop_last: True
-  collate_fn:
-    type: BatchImageCollateFunction
-
-
-val_dataloader:
-  type: DataLoader
-  dataset:
-    type: CocoDetection
-    img_folder: /data/yourdataset/val
-    ann_file: /data/yourdataset/val/val.json
-    return_masks: False
-    transforms:
-      type: Compose
-      ops: ~
-  shuffle: False
-  num_workers: 4
-  drop_last: False
-  collate_fn:
-    type: BatchImageCollateFunction
+task: detection
+
+evaluator:
+  type: CocoEvaluator
+  iou_types: ['bbox', ]
+
+num_classes: 3 # your dataset classes
+remap_mscoco_category: False
+
+train_dataloader:
+  type: DataLoader
+  dataset:
+    type: CocoDetection
+    img_folder: ./laydoc/train_images
+    ann_file: ./laydoc/annotations/train.json
+    return_masks: False
+    transforms:
+      type: Compose
+      ops: ~
+  shuffle: True
+  num_workers: 8
+  drop_last: True
+  collate_fn:
+    type: BatchImageCollateFunction
+
+
+val_dataloader:
+  type: DataLoader
+  dataset:
+    type: CocoDetection
+    img_folder: ./laydoc/val_images
+    ann_file: ./laydoc/annotations/val.json
+    return_masks: False
+    transforms:
+      type: Compose
+      ops: ~
+  shuffle: False
+  num_workers: 8
+  drop_last: False
+  collate_fn:
+    type: BatchImageCollateFunction