We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6144a50 commit c03f28bCopy full SHA for c03f28b
tools/infer/text/predict_from_yaml.py
@@ -226,8 +226,8 @@ def predict_single_step(cfg):
226
def predict_system(args, det_cfg, rec_cfg):
227
"""Run predict for both det and rec task"""
228
# merge image_dir option in model config
229
- det_cfg.predict.dataset_root = ""
230
- det_cfg.predict.data_dir = args.image_dir
+ det_cfg.predict.dataset.dataset_root = ""
+ det_cfg.predict.dataset.data_dir = args.image_dir
231
output_save_dir = det_cfg.predict.output_save_dir or "./output"
232
233
# get det result from predict
0 commit comments