From e19731a36a9f5bc0c7881ec0da5fd8e606cb3a3b Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Mon, 13 Jun 2022 20:00:48 +0800
Subject: [PATCH] fix traffic sign yaml (#2040)

---
 ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0.yaml              | 2 +-
 ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0_search.yaml       | 3 +--
 .../traffic_sign/SwinTransformer_tiny_patch4_window7_224.yaml  | 2 +-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0.yaml b/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0.yaml
index 4cc98b311..5362d07b7 100644
--- a/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0.yaml
+++ b/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0.yaml
@@ -54,7 +54,7 @@ DataLoader:
     dataset:
       name: ImageNetDataset
       image_root: ./dataset/
-      cls_label_path: ./dataset/tt100k_clas_v2/label_list_train.txt
+      cls_label_path: ./dataset/traffic_sign/label_list_train.txt
       delimiter: "\t"
       transform_ops:
         - DecodeImage:
diff --git a/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0_search.yaml b/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0_search.yaml
index bbac87a53..27fbc4b86 100644
--- a/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0_search.yaml
+++ b/ppcls/configs/PULC/traffic_sign/PPLCNet_x1_0_search.yaml
@@ -118,8 +118,7 @@ DataLoader:
       use_shared_memory: True
 
 Infer:
-  # infer_imgs: dataset/traffic_sign_demo/
-  infer_imgs: dataset/tt100k_clas_v2/test/
+  infer_imgs: deploy/images/PULC/traffic_sign/99603_17806.jpg
   batch_size: 10
   transforms:
     - DecodeImage:
diff --git a/ppcls/configs/PULC/traffic_sign/SwinTransformer_tiny_patch4_window7_224.yaml b/ppcls/configs/PULC/traffic_sign/SwinTransformer_tiny_patch4_window7_224.yaml
index 680ffe357..ae86ae622 100644
--- a/ppcls/configs/PULC/traffic_sign/SwinTransformer_tiny_patch4_window7_224.yaml
+++ b/ppcls/configs/PULC/traffic_sign/SwinTransformer_tiny_patch4_window7_224.yaml
@@ -113,7 +113,7 @@ DataLoader:
     dataset: 
       name: ImageNetDataset
       image_root: ./dataset/
-      cls_label_path: ./dataset/tt100k_clas_v2/label_list_test.txt
+      cls_label_path: ./dataset/traffic_sign/label_list_test.txt
       delimiter: "\t"
       transform_ops:
         - DecodeImage: