提交 1dacdfa7 编写于 作者: C cuicheng01

Update product configs

上级 38201d26
Global:
rec_inference_model_dir: "./inshop/rec/"
batch_size: 1
use_gpu: True
enable_mkldnn: True
cpu_num_threads: 100
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_path: "./inshop/inshop_index/"
image_root: "./inshop/dataset/"
data_file: "./inshop/inshop_gallery_demo.txt"
delimiter: " "
dist_type: "IP"
pq_size: 100
embedding_size: 512
Global:
infer_imgs: "./inshop/demo/01_3_back.jpg"
det_inference_model_dir: "./inshop/det/"
rec_inference_model_dir: "./inshop/rec/"
batch_size: 1
image_shape: [3, 640, 640]
threshold: 0.0
max_det_results: 3
labe_list:
- foreground
# inference engine config
use_gpu: True
enable_mkldnn: True
cpu_num_threads: 100
enable_benchmark: True
use_fp16: False
ir_optim: True
use_tensorrt: False
gpu_mem: 8000
enable_profile: False
DetPreProcess:
transform_ops:
- DetResize:
interp: 2
keep_ratio: false
target_size: [640, 640]
- DetNormalizeImage:
is_scale: true
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
- DetPermute: {}
DetPostProcess: {}
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_path: "./inshop/inshop_index"
search_budget: 100
return_k: 10
dist_type: "IP"
...@@ -77,7 +77,7 @@ DataLoader: ...@@ -77,7 +77,7 @@ DataLoader:
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Eval: Eval:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
dataset: dataset:
...@@ -101,7 +101,7 @@ DataLoader: ...@@ -101,7 +101,7 @@ DataLoader:
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Metric: Metric:
Train: Train:
- TopkAcc: - TopkAcc:
......
...@@ -34,7 +34,9 @@ Arch: ...@@ -34,7 +34,9 @@ Arch:
class_num: 3997 class_num: 3997
margin: 0.15 margin: 0.15
scale: 30 scale: 30
infer_output_key: "features"
infer_add_softmax: False
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
Train: Train:
...@@ -93,7 +95,7 @@ DataLoader: ...@@ -93,7 +95,7 @@ DataLoader:
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Eval: Eval:
Query: Query:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
...@@ -116,7 +118,7 @@ DataLoader: ...@@ -116,7 +118,7 @@ DataLoader:
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Gallery: Gallery:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
...@@ -139,7 +141,7 @@ DataLoader: ...@@ -139,7 +141,7 @@ DataLoader:
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Metric: Metric:
Eval: Eval:
......
...@@ -34,7 +34,9 @@ Arch: ...@@ -34,7 +34,9 @@ Arch:
class_num: 11319 class_num: 11319
margin: 0.15 margin: 0.15
scale: 30 scale: 30
infer_output_key: "features"
infer_add_softmax: False
# loss function config for traing/eval process # loss function config for traing/eval process
Loss: Loss:
Train: Train:
...@@ -93,7 +95,7 @@ DataLoader: ...@@ -93,7 +95,7 @@ DataLoader:
shuffle: True shuffle: True
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Eval: Eval:
Query: Query:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
...@@ -116,7 +118,7 @@ DataLoader: ...@@ -116,7 +118,7 @@ DataLoader:
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Gallery: Gallery:
# TOTO: modify to the latest trainer # TOTO: modify to the latest trainer
...@@ -139,7 +141,7 @@ DataLoader: ...@@ -139,7 +141,7 @@ DataLoader:
shuffle: False shuffle: False
loader: loader:
num_workers: 6 num_workers: 6
use_shared_memory: False use_shared_memory: True
Metric: Metric:
Eval: Eval:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册