Merge pull request #1528 from Intsigstephon/merge_configs
merge build_**.yaml and inference_**.yamlpull/1532/head
commit
098a934f50
|
@ -1,38 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/cartoon_rec_ResNet50_iCartoon_v1.0_infer/"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_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: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_cartoon/index/"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_cartoon/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_cartoon/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 2048
|
|
@ -1,36 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/general_PPLCNet_x2_5_lite_v1.0_infer"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: False
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: False
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
size: 224
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./drink_dataset/index"
|
||||
image_root: "./drink_dataset"
|
||||
data_file: "./drink_dataset/drink_label.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: " "
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
|
@ -1,36 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/general_PPLCNet_x2_5_lite_v1.0_infer"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
size: 224
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
image_root: "./drink_dataset_v1.0/gallery/"
|
||||
index_dir: "./drink_dataset_v1.0/index"
|
||||
data_file: "./drink_dataset_v1.0/gallery/drink_label.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
|
@ -1,36 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/logo_rec_ResNet50_Logo3K_v1.0_infer/"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
size: 224
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_logo/index/"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_logo/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_logo/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
|
@ -1,36 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/product_ResNet50_vd_aliproduct_v1.0_infer"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
size: 224
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_product/index"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_product/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_product/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
|
@ -1,39 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/product_MV3_x1_0_aliproduct_bin_v1.0_infer"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
size: 224
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess:
|
||||
main_indicator: Binarize
|
||||
Binarize:
|
||||
method: "round"
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "Flat" # supported: HNSW32, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_product/index_binary"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_product/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_product/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "hamming"
|
||||
embedding_size: 512
|
|
@ -1,36 +0,0 @@
|
|||
Global:
|
||||
rec_inference_model_dir: "./models/vehicle_cls_ResNet50_CompCars_v1.0_infer/"
|
||||
batch_size: 32
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
size: 224
|
||||
- NormalizeImage:
|
||||
scale: 0.00392157
|
||||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_vehicle/index/"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_vehicle/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_vehicle/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
|
@ -50,7 +50,16 @@ RecPreProcess:
|
|||
|
||||
RecPostProcess: null
|
||||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_cartoon/index/"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_cartoon/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_cartoon/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 2048
|
||||
batch_size: 32
|
||||
return_k: 5
|
||||
score_thres: 0.5
|
||||
score_thres: 0.5
|
|
@ -11,6 +11,7 @@ Global:
|
|||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
PreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
|
@ -24,10 +25,11 @@ PreProcess:
|
|||
order: ''
|
||||
channel_num: 3
|
||||
- ToCHWImage:
|
||||
|
||||
PostProcess:
|
||||
main_indicator: Topk
|
||||
Topk:
|
||||
topk: 5
|
||||
class_id_map_file: "../ppcls/utils/imagenet1k_label_list.txt"
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
save_dir: ./pre_label/
|
|
@ -11,6 +11,7 @@ Global:
|
|||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
PreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
|
@ -24,10 +25,11 @@ PreProcess:
|
|||
order: ''
|
||||
channel_num: 4
|
||||
- ToCHWImage:
|
||||
|
||||
PostProcess:
|
||||
main_indicator: Topk
|
||||
Topk:
|
||||
topk: 5
|
||||
class_id_map_file: "../ppcls/utils/imagenet1k_label_list.txt"
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
save_dir: ./pre_label/
|
|
@ -11,6 +11,7 @@ Global:
|
|||
use_tensorrt: False
|
||||
gpu_mem: 8000
|
||||
enable_profile: False
|
||||
|
||||
PreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
|
@ -24,10 +25,11 @@ PreProcess:
|
|||
order: ''
|
||||
channel_num: 3
|
||||
- ToCHWImage:
|
||||
|
||||
PostProcess:
|
||||
main_indicator: MultiLabelTopk
|
||||
MultiLabelTopk:
|
||||
topk: 5
|
||||
class_id_map_file: None
|
||||
SavePreLabel:
|
||||
save_dir: ./pre_label/
|
||||
save_dir: ./pre_label/
|
|
@ -30,4 +30,5 @@ DetPreProcess:
|
|||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
- DetPermute: {}
|
||||
DetPostProcess: {}
|
||||
|
||||
DetPostProcess: {}
|
|
@ -1,9 +1,9 @@
|
|||
Global:
|
||||
infer_imgs: "./recognition_demo_data_v1.2/test_drinks/hongniu_1.jpg"
|
||||
infer_imgs: "./drink_dataset_v1.0/test_images/hongniu_1.jpg"
|
||||
det_inference_model_dir: "./models/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer"
|
||||
rec_inference_model_dir: "./models/general_PPLCNet_x2_5_lite_v1.0_infer"
|
||||
rec_nms_thresold: 0.05
|
||||
|
||||
|
||||
batch_size: 1
|
||||
image_shape: [3, 640, 640]
|
||||
threshold: 0.2
|
||||
|
@ -11,11 +11,10 @@ Global:
|
|||
labe_list:
|
||||
- foreground
|
||||
|
||||
# inference engine config
|
||||
use_gpu: True
|
||||
enable_mkldnn: False
|
||||
cpu_num_threads: 10
|
||||
enable_benchmark: True
|
||||
enable_benchmark: False
|
||||
use_fp16: False
|
||||
ir_optim: True
|
||||
use_tensorrt: False
|
||||
|
@ -33,6 +32,7 @@ DetPreProcess:
|
|||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
- DetPermute: {}
|
||||
|
||||
DetPostProcess: {}
|
||||
|
||||
RecPreProcess:
|
||||
|
@ -50,6 +50,14 @@ RecPostProcess: null
|
|||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_dir: "./drink_dataset/index"
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./drink_dataset_v1.0/gallery"
|
||||
image_root: "./drink_dataset_v1.0/index"
|
||||
data_file: "./drink_dataset_v1.0/gallery/drink_label.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: " "
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
||||
batch_size: 32
|
||||
return_k: 5
|
||||
score_thres: 0.4
|
||||
score_thres: 0.4
|
|
@ -3,7 +3,7 @@ Global:
|
|||
det_inference_model_dir: "./models/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer"
|
||||
rec_inference_model_dir: "./models/general_PPLCNet_x2_5_lite_v1.0_infer"
|
||||
rec_nms_thresold: 0.05
|
||||
|
||||
|
||||
batch_size: 1
|
||||
image_shape: [3, 640, 640]
|
||||
threshold: 0.2
|
||||
|
@ -11,7 +11,6 @@ Global:
|
|||
labe_list:
|
||||
- foreground
|
||||
|
||||
# inference engine config
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
|
@ -33,6 +32,7 @@ DetPreProcess:
|
|||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
- DetPermute: {}
|
||||
|
||||
DetPostProcess: {}
|
||||
|
||||
RecPreProcess:
|
||||
|
@ -50,6 +50,14 @@ RecPostProcess: null
|
|||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_dir: "./drink_dataset_v1.0/index/"
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
image_root: "./drink_dataset_v1.0/gallery/"
|
||||
index_dir: "./drink_dataset_v1.0/index"
|
||||
data_file: "./drink_dataset_v1.0/gallery/drink_label.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
||||
batch_size: 32
|
||||
return_k: 5
|
||||
score_thres: 0.5
|
||||
score_thres: 0.5
|
|
@ -11,7 +11,6 @@ Global:
|
|||
labe_list:
|
||||
- foreground
|
||||
|
||||
# inference engine config
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
|
@ -33,6 +32,7 @@ DetPreProcess:
|
|||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
- DetPermute: {}
|
||||
|
||||
DetPostProcess: {}
|
||||
|
||||
RecPreProcess:
|
||||
|
@ -50,6 +50,14 @@ RecPostProcess: null
|
|||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_logo/index/"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_logo/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_logo/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
||||
batch_size: 32
|
||||
return_k: 5
|
||||
score_thres: 0.5
|
||||
score_thres: 0.5
|
|
@ -33,6 +33,7 @@ DetPreProcess:
|
|||
mean: [0.485, 0.456, 0.406]
|
||||
std: [0.229, 0.224, 0.225]
|
||||
- DetPermute: {}
|
||||
|
||||
DetPostProcess: {}
|
||||
|
||||
RecPreProcess:
|
||||
|
@ -50,6 +51,14 @@ RecPostProcess: null
|
|||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_product/index"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_product/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_product/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
||||
batch_size: 32
|
||||
return_k: 5
|
||||
score_thres: 0.5
|
||||
score_thres: 0.5
|
|
@ -53,8 +53,15 @@ RecPostProcess:
|
|||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
binary_index: true
|
||||
index_method: "Flat" # supported: HNSW32, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_product/index_binary"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_product/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_product/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "hamming"
|
||||
embedding_size: 512
|
||||
batch_size: 32
|
||||
binary_index: true
|
||||
return_k: 5
|
||||
score_thres: 0
|
||||
|
||||
score_thres: 0
|
|
@ -2,13 +2,6 @@ Global:
|
|||
infer_imgs: "./images/wangzai.jpg"
|
||||
rec_inference_model_dir: "./models/product_ResNet50_vd_aliproduct_v1.0_infer"
|
||||
batch_size: 1
|
||||
image_shape: [3, 640, 640]
|
||||
threshold: 0.5
|
||||
max_det_results: 1
|
||||
labe_list:
|
||||
- foreground
|
||||
|
||||
# inference engine config
|
||||
use_gpu: False
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
|
@ -32,4 +25,4 @@ RecPreProcess:
|
|||
order: ''
|
||||
- ToCHWImage:
|
||||
|
||||
RecPostProcess: null
|
||||
RecPostProcess: null
|
|
@ -11,7 +11,6 @@ Global:
|
|||
labe_list:
|
||||
- foreground
|
||||
|
||||
# inference engine config
|
||||
use_gpu: True
|
||||
enable_mkldnn: True
|
||||
cpu_num_threads: 10
|
||||
|
@ -36,7 +35,6 @@ DetPreProcess:
|
|||
|
||||
DetPostProcess: {}
|
||||
|
||||
|
||||
RecPreProcess:
|
||||
transform_ops:
|
||||
- ResizeImage:
|
||||
|
@ -52,6 +50,14 @@ RecPostProcess: null
|
|||
|
||||
# indexing engine config
|
||||
IndexProcess:
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_vehicle/index/"
|
||||
return_k: 5
|
||||
score_thres: 0.5
|
||||
index_method: "HNSW32" # supported: HNSW32, IVF, Flat
|
||||
index_dir: "./recognition_demo_data_v1.1/gallery_vehicle/index/"
|
||||
image_root: "./recognition_demo_data_v1.1/gallery_vehicle/"
|
||||
data_file: "./recognition_demo_data_v1.1/gallery_vehicle/data_file.txt"
|
||||
index_operation: "new" # suported: "append", "remove", "new"
|
||||
delimiter: "\t"
|
||||
dist_type: "IP"
|
||||
embedding_size: 512
|
||||
batch_size: 32
|
||||
return_k: 5
|
||||
score_thres: 0.5
|
|
@ -38,10 +38,10 @@
|
|||
|
||||
<a name="轻量级通用主体检测模型与轻量级通用识别模型"></a>
|
||||
|
||||
| 模型简介 | 推荐场景 | inference 模型 | 预测配置文件 | 构建索引库的配置文件 |
|
||||
| ------------ | ------------- | -------- | ------- | -------- |
|
||||
| 轻量级通用主体检测模型 | 通用场景 |[tar 格式文件下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.tar) [zip 格式文件下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.zip) | - | - |
|
||||
| 轻量级通用识别模型 | 通用场景 | [tar 格式下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/general_PPLCNet_x2_5_lite_v1.0_infer.tar) [zip 格式文件下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/general_PPLCNet_x2_5_lite_v1.0_infer.zip) | [inference_general.yaml](../../../deploy/configs/inference_general.yaml) | [build_general.yaml](../../../deploy/configs/build_general.yaml) |
|
||||
| 模型简介 | 推荐场景 | inference 模型 | 预测配置文件 |
|
||||
| ------------ | ------------- | -------- | ------- |
|
||||
| 轻量级通用主体检测模型 | 通用场景 |[tar 格式文件下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.tar) [zip 格式文件下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/picodet_PPLCNet_x2_5_mainbody_lite_v1.0_infer.zip) | - |
|
||||
| 轻量级通用识别模型 | 通用场景 | [tar 格式下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/general_PPLCNet_x2_5_lite_v1.0_infer.tar) [zip 格式文件下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/general_PPLCNet_x2_5_lite_v1.0_infer.zip) | [inference_general.yaml](../../../deploy/configs/inference_general.yaml) |
|
||||
|
||||
注意:由于部分解压缩软件在解压上述 `tar` 格式文件时存在问题,建议非命令行用户下载 `zip` 格式文件并解压。`tar` 格式文件建议使用命令 `tar xf xxx.tar` 解压。
|
||||
|
||||
|
@ -129,7 +129,7 @@ wget https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/data/drink_da
|
|||
|
||||
```shell
|
||||
# 下面是使用下载的服务端商品识别模型进行索引库构建
|
||||
python3.7 python/build_gallery.py -c configs/build_general.yaml -o Global.rec_inference_model_dir=./models/general_PPLCNet_x2_5_lite_v1.0_infer
|
||||
python3.7 python/build_gallery.py -c configs/inference_general.yaml -o Global.rec_inference_model_dir=./models/general_PPLCNet_x2_5_lite_v1.0_infer
|
||||
```
|
||||
|
||||
<a name="瓶装饮料识别与检索"></a>
|
||||
|
@ -256,7 +256,7 @@ python3.7 python/predict_system.py -c configs/inference_general.yaml -o Global.i
|
|||
使用下面的命令构建 `index` 索引,加速识别后的检索过程。
|
||||
|
||||
```shell
|
||||
python3.7 python/build_gallery.py -c configs/build_general.yaml -o IndexProcess.data_file="./drink_dataset_v1.0/gallery/drink_label_all.txt" -o IndexProcess.index_dir="./drink_dataset_v1.0/index_all"
|
||||
python3.7 python/build_gallery.py -c configs/inference_general.yaml -o IndexProcess.data_file="./drink_dataset_v1.0/gallery/drink_label_all.txt" -o IndexProcess.index_dir="./drink_dataset_v1.0/index_all"
|
||||
```
|
||||
|
||||
最终新的索引信息保存在文件夹 `./drink_dataset_v1.0/index_all` 中。具体 `yaml` 请参考[向量检索文档](../image_recognition_pipeline/vector_search.md)。
|
||||
|
@ -290,14 +290,14 @@ python3.7 python/predict_system.py -c configs/inference_general.yaml -o Global.i
|
|||
|
||||
目前,我们更推荐您使用[轻量级通用主体检测模型与轻量级通用识别模型](#轻量级通用主体检测模型与轻量级通用识别模型),以获得更好的测试结果。但是如果您希望体验服务端识别模型,服务器端通用主体检测模型与各方向识别模型、测试数据下载地址以及对应的配置文件地址如下。
|
||||
|
||||
| 模型简介 | 推荐场景 | inference 模型 | 预测配置文件 | 构建索引库的配置文件 |
|
||||
| ------------ | ------------- | -------- | ------- | -------- |
|
||||
| 通用主体检测模型 | 通用场景 |[模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/ppyolov2_r50vd_dcn_mainbody_v1.0_infer.tar) | - | - |
|
||||
| Logo 识别模型 | Logo 场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/logo_rec_ResNet50_Logo3K_v1.0_infer.tar) | [inference_logo.yaml](../../../deploy/configs/inference_logo.yaml) | [build_logo.yaml](../../../deploy/configs/build_logo.yaml) |
|
||||
| 动漫人物识别模型 | 动漫人物场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/cartoon_rec_ResNet50_iCartoon_v1.0_infer.tar) | [inference_cartoon.yaml](../../../deploy/configs/inference_cartoon.yaml) | [build_cartoon.yaml](../../../deploy/configs/build_cartoon.yaml) |
|
||||
| 车辆细分类模型 | 车辆场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/vehicle_cls_ResNet50_CompCars_v1.0_infer.tar) | [inference_vehicle.yaml](../../../deploy/configs/inference_vehicle.yaml) | [build_vehicle.yaml](../../../deploy/configs/build_vehicle.yaml) |
|
||||
| 商品识别模型 | 商品场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/product_ResNet50_vd_aliproduct_v1.0_infer.tar) | [inference_product.yaml](../../../deploy/configs/inference_product.yaml) | [build_product.yaml](../../../deploy/configs/build_product.yaml) |
|
||||
| 车辆 ReID 模型 | 车辆 ReID 场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/vehicle_reid_ResNet50_VERIWild_v1.0_infer.tar) | - | - |
|
||||
| 模型简介 | 推荐场景 | inference 模型 | 预测配置文件 |
|
||||
| ------------ | ------------- | -------- | ------- |
|
||||
| 通用主体检测模型 | 通用场景 |[模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/ppyolov2_r50vd_dcn_mainbody_v1.0_infer.tar) | - |
|
||||
| Logo 识别模型 | Logo 场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/logo_rec_ResNet50_Logo3K_v1.0_infer.tar) | [inference_logo.yaml](../../../deploy/configs/inference_logo.yaml) |
|
||||
| 动漫人物识别模型 | 动漫人物场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/cartoon_rec_ResNet50_iCartoon_v1.0_infer.tar) | [inference_cartoon.yaml](../../../deploy/configs/inference_cartoon.yaml) |
|
||||
| 车辆细分类模型 | 车辆场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/vehicle_cls_ResNet50_CompCars_v1.0_infer.tar) | [inference_vehicle.yaml](../../../deploy/configs/inference_vehicle.yaml) |
|
||||
| 商品识别模型 | 商品场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/product_ResNet50_vd_aliproduct_v1.0_infer.tar) | [inference_product.yaml](../../../deploy/configs/inference_product.yaml) |
|
||||
| 车辆 ReID 模型 | 车辆 ReID 场景 | [模型下载链接](https://paddle-imagenet-models-name.bj.bcebos.com/dygraph/rec/models/inference/vehicle_reid_ResNet50_VERIWild_v1.0_infer.tar) | [inference_vehicle.yaml](../../../deploy/configs/inference_vehicle.yaml) |
|
||||
|
||||
```shell
|
||||
cd PaddleClas/deploy/
|
||||
|
|
Loading…
Reference in New Issue