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" ]