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
|
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:
|
||||||
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:
|
||||||
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:
|
||||||
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:
|
||||||
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:
|
||||||
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:
|
||||||
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:
|
||||||
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:
|
||||||
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:
|
||||||
shuffle: False
|
shuffle: False
|
||||||
loader:
|
loader:
|
||||||
num_workers: 6
|
num_workers: 6
|
||||||
use_shared_memory: False
|
use_shared_memory: True
|
||||||
|
|
||||||
Metric:
|
Metric:
|
||||||
Eval:
|
Eval:
|
||||||
|
|
Loading…
Reference in New Issue