add fleet chain and update inference_rec.yaml and corresponding output for feature_extraction.md
parent
4620cfb9bc
commit
c449ba3420
|
@ -15,14 +15,15 @@ Global:
|
|||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
resize_short: 256
|
||||
- CropImage:
|
||||
size: 224
|
||||
size: [224, 224]
|
||||
return_numpy: False
|
||||
interpolation: bilinear
|
||||
backend: cv2
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
scale: 1.0/255.0
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ""
|
||||
order: hwc
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
|
|
@ -252,12 +252,12 @@ python3.7 python/predict_rec.py \
|
|||
得到的特征输出格式如下所示:
|
||||
|
||||
```log
|
||||
wangzai.jpg: [-4.72979844e-02 3.40038240e-02 -4.06982675e-02 2.46225717e-03
|
||||
9.67078656e-03 2.70162839e-02 -8.85589980e-03 -4.71983254e-02
|
||||
6.18615076e-02 1.38509814e-02 -1.11342799e-02 6.73768669e-02
|
||||
-1.03065455e-02 -4.88462113e-02 8.46387446e-03 1.57074817e-02
|
||||
wangzai.jpg: [-7.82453567e-02 2.55877394e-02 -3.66694555e-02 1.34572461e-02
|
||||
4.39076796e-02 -2.34078392e-02 -9.49947070e-03 1.28221214e-02
|
||||
5.53947650e-02 1.01355985e-02 -1.06436480e-02 4.97181974e-02
|
||||
-2.21862812e-02 -1.75557341e-02 1.55848479e-02 -3.33278324e-03
|
||||
...
|
||||
-3.14170569e-02 7.35917836e-02 -3.09373233e-02 -2.31755469e-02]
|
||||
-3.40284109e-02 8.35561901e-02 2.10910216e-02 -3.27066667e-02]
|
||||
```
|
||||
|
||||
在实际使用过程中,仅仅得到特征可能并不能满足业务需求。如果想进一步通过特征检索来进行图像识别,可以参照文档 [向量检索](./vector_search.md)。
|
||||
|
|
|
@ -0,0 +1,54 @@
|
|||
===========================train_params===========================
|
||||
model_name:GeneralRecognition_PPLCNet_x2_5
|
||||
python:python3.7
|
||||
gpu_list:192.168.0.1,192.168.0.2;0,1
|
||||
-o Global.device:gpu
|
||||
-o Global.auto_cast:null
|
||||
-o Global.epochs:lite_train_lite_infer=2|whole_train_whole_infer=120
|
||||
-o Global.output_dir:./output/
|
||||
-o DataLoader.Train.sampler.batch_size:8
|
||||
-o Global.pretrained_model:null
|
||||
train_model_name:latest
|
||||
train_infer_img_dir:./dataset/ILSVRC2012/val
|
||||
null:null
|
||||
##
|
||||
trainer:norm_train
|
||||
norm_train:tools/train.py -c ppcls/configs/GeneralRecognitionV2/GeneralRecognitionV2_PPLCNetV2_base.yaml -o Global.seed=1234 -o DataLoader.Train.sampler.shuffle=False -o DataLoader.Train.loader.num_workers=0 -o DataLoader.Train.loader.use_shared_memory=False -o Global.eval_during_train=False -o Global.save_interval=2 -o DataLoader.Train.dataset.cls_label_path=./dataset/train_reg_all_data.txt -o DataLoader.Train.loader.sampler.batch_size=8
|
||||
pact_train:null
|
||||
fpgm_train:null
|
||||
distill_train:null
|
||||
null:null
|
||||
null:null
|
||||
##
|
||||
===========================eval_params===========================
|
||||
eval:tools/eval.py -c ppcls/configs/GeneralRecognitionV2/GeneralRecognitionV2_PPLCNetV2_base.yaml
|
||||
null:null
|
||||
##
|
||||
===========================infer_params==========================
|
||||
-o Global.save_inference_dir:./inference
|
||||
-o Global.pretrained_model:
|
||||
norm_export:tools/export_model.py -c ppcls/configs/GeneralRecognitionV2/GeneralRecognitionV2_PPLCNetV2_base.yaml
|
||||
quant_export:null
|
||||
fpgm_export:null
|
||||
distill_export:null
|
||||
kl_quant:null
|
||||
export2:null
|
||||
pretrained_model_url:https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/pretrain/PPShiTuV2/general_PPLCNetV2_base_pretrained_v1.0.pdparams
|
||||
infer_model:../inference/
|
||||
infer_export:True
|
||||
infer_quant:Fasle
|
||||
inference:python/predict_rec.py -c configs/inference_rec.yaml
|
||||
-o Global.use_gpu:True|False
|
||||
-o Global.enable_mkldnn:False
|
||||
-o Global.cpu_num_threads:1
|
||||
-o Global.batch_size:1
|
||||
-o Global.use_tensorrt:False
|
||||
-o Global.use_fp16:False
|
||||
-o Global.rec_inference_model_dir:../inference
|
||||
-o Global.infer_imgs:../dataset/Aliproduct/demo_test/
|
||||
-o Global.save_log_path:null
|
||||
-o Global.benchmark:False
|
||||
null:null
|
||||
null:null
|
||||
===========================infer_benchmark_params==========================
|
||||
random_infer_input:[{float32,[3,224,224]}]
|
Loading…
Reference in New Issue