From f60c6a043d147bbdb214e88adaa4a80ed8218ee3 Mon Sep 17 00:00:00 2001
From: xiaoluomi <49263480+xiaoluomi@users.noreply.github.com>
Date: Mon, 8 May 2023 14:25:54 +0800
Subject: [PATCH] add rtdetr_ptq_demo (#1745)

---
 .../detection/configs/rtdetr_ptq.yaml         | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 example/post_training_quantization/detection/configs/rtdetr_ptq.yaml

diff --git a/example/post_training_quantization/detection/configs/rtdetr_ptq.yaml b/example/post_training_quantization/detection/configs/rtdetr_ptq.yaml
new file mode 100644
index 00000000..c52864de
--- /dev/null
+++ b/example/post_training_quantization/detection/configs/rtdetr_ptq.yaml
@@ -0,0 +1,34 @@
+
+input_list: ['im_shape', 'image', 'scale_factor']
+model_dir: ./rtdetr_r50vd_6x_coco
+model_filename: model.pdmodel
+params_filename: model.pdiparams
+metric: COCO
+num_classes: 80
+  
+
+# Datset configuration
+TrainDataset:
+  !COCODataSet
+    image_dir: train2017
+    anno_path: annotations/instances_train2017.json
+    dataset_dir: /dataset/coco/
+
+EvalDataset:
+  !COCODataSet
+    image_dir: val2017
+    anno_path: annotations/instances_val2017.json
+    dataset_dir: /dataset/coco/
+
+worker_num: 0
+
+# preprocess reader in test
+EvalReader:
+  sample_transforms:
+    - Decode: {}
+    - Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
+    - NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
+    - Permute: {}
+  batch_size: 8
+  shuffle: false
+  drop_last: false
-- 
GitLab