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

38 lines
946 B
YAML
Raw Normal View History

2022-03-13 16:42:16 +08:00
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"
2022-03-14 21:30:57 +08:00
to_model_names:
image: inputs
2022-03-15 15:41:42 +08:00
from_model_indexes:
2022-03-14 21:30:57 +08:00
logits: 0
2022-03-13 16:42:16 +08:00
- name: TopK
type: postprocessor
k: 10
2022-03-15 15:41:42 +08:00
class_id_map_file: "../../../ppcls/utils/imagenet1k_label_list.txt"
2022-03-13 16:42:16 +08:00
save_dir: None