From 2796ad1cd7b63b3c5d9a7a8af5a38174e9e89c04 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Thu, 30 Jun 2022 19:07:07 +0800
Subject: [PATCH] fix cfg

---
 .../PPLCNet_x1_0_distillation.yaml            | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/ppcls/configs/PULC/vehicle_attribute/PPLCNet_x1_0_distillation.yaml b/ppcls/configs/PULC/vehicle_attribute/PPLCNet_x1_0_distillation.yaml
index c5144f373..208eb00d0 100644
--- a/ppcls/configs/PULC/vehicle_attribute/PPLCNet_x1_0_distillation.yaml
+++ b/ppcls/configs/PULC/vehicle_attribute/PPLCNet_x1_0_distillation.yaml
@@ -142,6 +142,26 @@ DataLoader:
       num_workers: 8
       use_shared_memory: True
 
+Infer:
+  infer_imgs: ./deploy/images/PULC/vehicle_attribute/0002_c002_00030670_0.jpg
+  batch_size: 10
+  transforms:
+    - DecodeImage:
+        to_rgb: True
+        channel_first: False
+    - ResizeImage:
+        size: [256, 192]
+    - NormalizeImage:
+        scale: 1.0/255.0
+        mean: [0.485, 0.456, 0.406]
+        std: [0.229, 0.224, 0.225]
+        order: ''
+    - ToCHWImage:
+  PostProcess:
+    name: VehicleAttribute
+    color_threshold: 0.5
+    type_threshold: 0.5
+
 
 Metric:
   Eval: