From c94796494f109cd7443f7f059a797cb8a41240ab Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Thu, 17 Jun 2021 03:12:22 +0000 Subject: [PATCH] Update product and cartoonface configs --- ppcls/configs/Cartoonface/ResNet50_icartoon.yaml | 2 +- ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml b/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml index 01ab83e1..bd39afd9 100644 --- a/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml +++ b/ppcls/configs/Cartoonface/ResNet50_icartoon.yaml @@ -14,7 +14,7 @@ Global: use_visualdl: False image_shape: [3, 224, 224] infer_imgs: - save_inference_dir: "./output" + save_inference_dir: "./inference" feature_normalize: True Arch: diff --git a/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml b/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml index 5c8a01ab..9a8469d4 100644 --- a/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml +++ b/ppcls/configs/Products/ResNet50_vd_Aliproduct.yaml @@ -19,6 +19,9 @@ Global: # model architecture Arch: name: RecModel + infer_output_key: features + infer_add_softmax: False + Backbone: name: ResNet50_vd pretrained: True -- GitLab