PaddleClas/deploy/python/ppshitu_v2/configs/test_cls_config.yaml

38 lines
946 B
YAML

Global:
Engine: POPEngine
infer_imgs: "../../images/wangzai.jpg"
Modules:
- name:
type: AlgoMod
processors:
- name: ImageProcessor
type: preprocessor
ops:
- ResizeImage:
resize_short: 256
- CropImage:
size: 224
- NormalizeImage:
scale: 0.00392157
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
order: hwc
- ToCHWImage:
- GetShapeInfo:
configs:
order: chw
- ToBatch:
- name: PaddlePredictor
type: predictor
inference_model_dir: "./MobileNetV2_infer"
to_model_names:
image: inputs
from_model_indexes:
logits: 0
- name: TopK
type: postprocessor
k: 10
class_id_map_file: "../../../ppcls/utils/imagenet1k_label_list.txt"
save_dir: None