Skip to content

Commit c03f28b

Browse files
authored
bug fix of online inference (#653)
1 parent 6144a50 commit c03f28b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tools/infer/text/predict_from_yaml.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -226,8 +226,8 @@ def predict_single_step(cfg):
226226
def predict_system(args, det_cfg, rec_cfg):
227227
"""Run predict for both det and rec task"""
228228
# merge image_dir option in model config
229-
det_cfg.predict.dataset_root = ""
230-
det_cfg.predict.data_dir = args.image_dir
229+
det_cfg.predict.dataset.dataset_root = ""
230+
det_cfg.predict.dataset.data_dir = args.image_dir
231231
output_save_dir = det_cfg.predict.output_save_dir or "./output"
232232

233233
# get det result from predict

0 commit comments

Comments
 (0)