更换文档检测模型

2024-08-27 14:42:45 +08:00
parent aea6f19951
commit 1514e09c40
2072 changed files with 254336 additions and 4967 deletions
--- a/paddle_detection/configs/smalldet/_base_/DOTA_sliced_500_025_detection.yml
+++ b/paddle_detection/configs/smalldet/_base_/DOTA_sliced_500_025_detection.yml
@@ -0,0 +1,20 @@
+metric: COCO
+num_classes: 15
+
+TrainDataset:
+  !COCODataSet
+    image_dir: train_images_500_025
+    anno_path: train_500_025.json
+    dataset_dir: dataset/dota_sliced
+    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
+
+EvalDataset:
+  !COCODataSet
+    image_dir: val_images_500_025
+    anno_path: val_500_025.json
+    dataset_dir: dataset/dota_sliced
+
+TestDataset:
+  !ImageFolder
+    anno_path: val_500_025.json
+    dataset_dir: dataset/dota_sliced
--- a/paddle_detection/configs/smalldet/_base_/visdrone_sliced_640_025_detection.yml
+++ b/paddle_detection/configs/smalldet/_base_/visdrone_sliced_640_025_detection.yml
@@ -0,0 +1,20 @@
+metric: COCO
+num_classes: 10
+
+TrainDataset:
+  !COCODataSet
+    image_dir: train_images_640_025
+    anno_path: train_640_025.json
+    dataset_dir: dataset/visdrone_sliced
+    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
+
+EvalDataset:
+  !COCODataSet
+    image_dir: val_images_640_025
+    anno_path: val_640_025.json
+    dataset_dir: dataset/visdrone_sliced
+
+TestDataset:
+  !ImageFolder
+    anno_path: val_640_025.json
+    dataset_dir: dataset/visdrone_sliced
--- a/paddle_detection/configs/smalldet/_base_/xview_sliced_400_025_detection.yml
+++ b/paddle_detection/configs/smalldet/_base_/xview_sliced_400_025_detection.yml
@@ -0,0 +1,20 @@
+metric: COCO
+num_classes: 60
+
+TrainDataset:
+  !COCODataSet
+    image_dir: train_images_400_025
+    anno_path: train_400_025.json
+    dataset_dir: dataset/xview_sliced
+    data_fields: ['image', 'gt_bbox', 'gt_class', 'is_crowd']
+
+EvalDataset:
+  !COCODataSet
+    image_dir: val_images_400_025
+    anno_path: val_400_025.json
+    dataset_dir: dataset/xview_sliced
+
+TestDataset:
+  !ImageFolder
+    anno_path: val_400_025.json
+    dataset_dir: dataset/xview_sliced