Skip to content

Commit f60c6a0

Browse files
authored
add rtdetr_ptq_demo (#1745)
1 parent 673c158 commit f60c6a0

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
2+
input_list: ['im_shape', 'image', 'scale_factor']
3+
model_dir: ./rtdetr_r50vd_6x_coco
4+
model_filename: model.pdmodel
5+
params_filename: model.pdiparams
6+
metric: COCO
7+
num_classes: 80
8+
9+
10+
# Datset configuration
11+
TrainDataset:
12+
!COCODataSet
13+
image_dir: train2017
14+
anno_path: annotations/instances_train2017.json
15+
dataset_dir: /dataset/coco/
16+
17+
EvalDataset:
18+
!COCODataSet
19+
image_dir: val2017
20+
anno_path: annotations/instances_val2017.json
21+
dataset_dir: /dataset/coco/
22+
23+
worker_num: 0
24+
25+
# preprocess reader in test
26+
EvalReader:
27+
sample_transforms:
28+
- Decode: {}
29+
- Resize: {target_size: [640, 640], keep_ratio: False, interp: 2}
30+
- NormalizeImage: {mean: [0., 0., 0.], std: [1., 1., 1.], norm_type: none}
31+
- Permute: {}
32+
batch_size: 8
33+
shuffle: false
34+
drop_last: false

0 commit comments

Comments
 (0)