From 9396b5bd8979b97ba9442a0befc2bfbabb51111a Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Fri, 10 Jun 2022 12:27:14 +0000 Subject: [PATCH] fix: car_exists -> vehicle_exists --- .../inference_vehicle_exists.yaml} | 8 ++++---- .../objects365_00001507.jpeg | Bin .../objects365_00001521.jpeg | Bin paddleclas.py | 2 +- 4 files changed, 5 insertions(+), 5 deletions(-) rename deploy/configs/PULC/{car_exists/inference_car_exists.yaml => vehicle_exists/inference_vehicle_exists.yaml} (77%) rename deploy/images/PULC/{car_exists => vehicle_exists}/objects365_00001507.jpeg (100%) rename deploy/images/PULC/{car_exists => vehicle_exists}/objects365_00001521.jpeg (100%) diff --git a/deploy/configs/PULC/car_exists/inference_car_exists.yaml b/deploy/configs/PULC/vehicle_exists/inference_vehicle_exists.yaml similarity index 77% rename from deploy/configs/PULC/car_exists/inference_car_exists.yaml rename to deploy/configs/PULC/vehicle_exists/inference_vehicle_exists.yaml index 7204b2723..cdd289dda 100644 --- a/deploy/configs/PULC/car_exists/inference_car_exists.yaml +++ b/deploy/configs/PULC/vehicle_exists/inference_vehicle_exists.yaml @@ -1,6 +1,6 @@ Global: - infer_imgs: "./images/PULC/car_exists/objects365_00001507.jpeg" - inference_model_dir: "./models/car_exists_infer" + infer_imgs: "./images/PULC/vehicle_exists/objects365_00001507.jpeg" + inference_model_dir: "./models/vehicle_exists_infer" batch_size: 1 use_gpu: True enable_mkldnn: False @@ -30,7 +30,7 @@ PostProcess: main_indicator: ThreshOutput ThreshOutput: threshold: 0.5 - label_0: nocar - label_1: contains_car + label_0: no_vehicle + label_1: contains_vehicle SavePreLabel: save_dir: ./pre_label/ diff --git a/deploy/images/PULC/car_exists/objects365_00001507.jpeg b/deploy/images/PULC/vehicle_exists/objects365_00001507.jpeg similarity index 100% rename from deploy/images/PULC/car_exists/objects365_00001507.jpeg rename to deploy/images/PULC/vehicle_exists/objects365_00001507.jpeg diff --git a/deploy/images/PULC/car_exists/objects365_00001521.jpeg b/deploy/images/PULC/vehicle_exists/objects365_00001521.jpeg similarity index 100% rename from deploy/images/PULC/car_exists/objects365_00001521.jpeg rename to deploy/images/PULC/vehicle_exists/objects365_00001521.jpeg diff --git a/paddleclas.py b/paddleclas.py index be9ebf5c1..8ab4eccbf 100644 --- a/paddleclas.py +++ b/paddleclas.py @@ -178,7 +178,7 @@ IMN_MODEL_SERIES = { PULC_MODEL_BASE_DOWNLOAD_URL = "https://paddleclas.bj.bcebos.com/models/PULC/{}_infer.tar" PULC_MODELS = [ "person_exists", "person_attribute", "safety_helmet", "traffic_sign", - "car_exists", "vehicle_attribute", "textline_orientation", + "vehicle_exists", "vehicle_attribute", "textline_orientation", "text_image_orientation", "language_classification" ]