diff --git a/ppcls/configs/Logo/ResNet50_ReID.yaml b/ppcls/configs/Logo/ResNet50_ReID.yaml index 43798a209df5333ce4f14e3614265e5db46a57a4..22b1850d2c1016957718f75a9cab9786f1e0e2ed 100644 --- a/ppcls/configs/Logo/ResNet50_ReID.yaml +++ b/ppcls/configs/Logo/ResNet50_ReID.yaml @@ -33,6 +33,7 @@ Arch: margin: 0.35 scale: 64 embedding_size: 512 + class_num: 3000 # loss function config for traing/eval process Loss: @@ -149,19 +150,3 @@ Metric: topk: [1, 5] - mAP: {} - -Infer: - infer_imgs: "docs/images/whl/demo.jpg" - batch_size: 10 - transforms: - - DecodeImage: - to_rgb: True - channel_first: False - - ResizeImage: - resize_short: 224 - - NormalizeImage: - scale: 1.0/255.0 - mean: [0.485, 0.456, 0.406] - std: [0.229, 0.224, 0.225] - order: '' - - ToCHWImage: