cheng-hust commited on
Commit
fd9f5d2
·
verified ·
1 Parent(s): a6c6c04

Upload coco_detectioncoco.yml

Browse files
configs/dataset/coco_detectioncoco.yml ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ task: detection
2
+
3
+ num_classes: 80 #aitod 9
4
+ remap_mscoco_category: False
5
+
6
+ train_dataloader:
7
+ type: DataLoader
8
+ dataset:
9
+ type: CocoDetection
10
+ img_folder: ../../AITOD/train/images/
11
+ #img_folder: /data/student1/keshe/RT-DETR/rtdetr_pytorch/sahi/sliced/aitod_val_v1_images_400_025/
12
+ ann_file: ../../AITOD/annotations/aitod_train_v1.json
13
+ # ann_file: /data/student1/keshe/RT-DETR/rtdetr_pytorch/sahi/sliced/aitod_val_v1_400_025.json
14
+ transforms:
15
+ type: Compose
16
+ ops: ~
17
+ shuffle: True
18
+ batch_size: 4
19
+ num_workers: 1
20
+ drop_last: True
21
+
22
+
23
+ val_dataloader:
24
+ type: DataLoader
25
+ dataset:
26
+ type: CocoDetection
27
+ img_folder: ../../AITOD/val/images/
28
+ #img_folder: /data/student1/keshe/RT-DETR/rtdetr_pytorch/sahi/sliced/aitod_val_v1_images_400_025/
29
+ ann_file: ../../AITOD/annotations/aitod_val_v1.json
30
+ #ann_file: /data/student1/keshe/RT-DETR/rtdetr_pytorch/sahi/sliced/aitod_val_v1_400_025.json
31
+ transforms:
32
+ type: Compose
33
+ ops: ~
34
+
35
+ shuffle: False
36
+ batch_size: 1
37
+ num_workers: 1
38
+ drop_last: False