commit
38f0f5f48c
|
@ -0,0 +1,35 @@
|
|||
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
|
|
@ -0,0 +1,54 @@
|
|||
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:
|
|||
shuffle: True
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
Eval:
|
||||
# TOTO: modify to the latest trainer
|
||||
dataset:
|
||||
|
@ -101,7 +101,7 @@ DataLoader:
|
|||
shuffle: False
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
Metric:
|
||||
Train:
|
||||
- TopkAcc:
|
||||
|
|
|
@ -34,6 +34,8 @@ Arch:
|
|||
class_num: 3997
|
||||
margin: 0.15
|
||||
scale: 30
|
||||
infer_output_key: "features"
|
||||
infer_add_softmax: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
@ -93,7 +95,7 @@ DataLoader:
|
|||
shuffle: True
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
Eval:
|
||||
Query:
|
||||
# TOTO: modify to the latest trainer
|
||||
|
@ -116,7 +118,7 @@ DataLoader:
|
|||
shuffle: False
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
|
||||
Gallery:
|
||||
# TOTO: modify to the latest trainer
|
||||
|
@ -139,7 +141,7 @@ DataLoader:
|
|||
shuffle: False
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
|
||||
Metric:
|
||||
Eval:
|
||||
|
|
|
@ -34,6 +34,8 @@ Arch:
|
|||
class_num: 11319
|
||||
margin: 0.15
|
||||
scale: 30
|
||||
infer_output_key: "features"
|
||||
infer_add_softmax: False
|
||||
|
||||
# loss function config for traing/eval process
|
||||
Loss:
|
||||
|
@ -93,7 +95,7 @@ DataLoader:
|
|||
shuffle: True
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
Eval:
|
||||
Query:
|
||||
# TOTO: modify to the latest trainer
|
||||
|
@ -116,7 +118,7 @@ DataLoader:
|
|||
shuffle: False
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
|
||||
Gallery:
|
||||
# TOTO: modify to the latest trainer
|
||||
|
@ -139,7 +141,7 @@ DataLoader:
|
|||
shuffle: False
|
||||
loader:
|
||||
num_workers: 6
|
||||
use_shared_memory: False
|
||||
use_shared_memory: True
|
||||
|
||||
Metric:
|
||||
Eval:
|
||||
|
|
Loading…
Reference in New Issue