generated from zhangwei/Matrixai
50 lines
1.6 KiB
YAML
50 lines
1.6 KiB
YAML
# 动态检测
|
|
move_model:
|
|
model_path: "./model/step0/step0.engine"
|
|
score_threshold: 0
|
|
class: []
|
|
# 关键区域识别
|
|
train_step1_model:
|
|
model_path: "./model/step1/step1.engine"
|
|
score_threshold: 0.6
|
|
class: [ "HEADNUM",
|
|
"PROPERTY",
|
|
"TYPENUM_K",
|
|
"TYPENUM_C",
|
|
"TYPENUM_P",
|
|
"TYPENUM_G",
|
|
"TYPENUM_N",
|
|
"CONTAINERNUM",
|
|
"TYPENUM_J",
|
|
"SPACE",
|
|
"SPACELK",
|
|
"SPACENX",
|
|
"SPACEG",
|
|
"SPACEP",
|
|
"SPACEJ",
|
|
"TYPENUM_W",
|
|
"SPACEW",
|
|
"SPACEM",
|
|
"SPACEU",
|
|
]
|
|
# 字符识别
|
|
train_step2_model:
|
|
model_path: "./model/step2/step2.engine"
|
|
score_threshold: 0.7
|
|
class: ["0","1","2","3","4","5","6","7","8","9",
|
|
"A","B","C","D","E","F","G","H","I","J","K","L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V","W", "X", "Y", "Z",
|
|
"change", "load", "m", "self", "t", "volume", "meter",")", "(", "?", "-"
|
|
]
|
|
# 集装箱关键区域识别
|
|
container_step1_model:
|
|
model_path: "./model/container_step1/con1.engine"
|
|
score_threshold: 0.6
|
|
class: ["CONTAINERNUM","CONTAINERNUM_REVERSE","SPACE_T","SPACENX_T","SPACEG_T","SPACEP_T","SPACEP_T"]
|
|
# 集装箱字符识别
|
|
container_step2_model:
|
|
model_path: "./model/container_step2/con2.engine"
|
|
score_threshold: 0.7
|
|
class: ["0","1","2","3","4","5","6","7","8","9",
|
|
"A","B","C","D","E","F","G","H","I","J","K","L", "M", "N", "O", "P", "Q", "R", "S", "T", "U", "V","W", "X", "Y", "Z",
|
|
"change", "load", "m", "self", "t", "volume", "meter",")", "(", "?", "-"
|
|
] |