mirror of https://github.com/open-mmlab/mmocr.git
[Fix] Fix browse_dataset (#1271)
parent
7cd96aaf79
commit
97f6c1d5d6
|
@ -3,7 +3,7 @@ import argparse
|
||||||
import os.path as osp
|
import os.path as osp
|
||||||
|
|
||||||
import mmcv
|
import mmcv
|
||||||
from mmcv import Config, DictAction
|
from mmengine import Config, DictAction
|
||||||
|
|
||||||
from mmocr.registry import DATASETS, VISUALIZERS
|
from mmocr.registry import DATASETS, VISUALIZERS
|
||||||
from mmocr.utils import register_all_modules
|
from mmocr.utils import register_all_modules
|
||||||
|
@ -64,7 +64,7 @@ def main():
|
||||||
visualizer.add_datasample(
|
visualizer.add_datasample(
|
||||||
name=osp.basename(img_path),
|
name=osp.basename(img_path),
|
||||||
image=img,
|
image=img,
|
||||||
gt_sample=data_sample,
|
data_sample=data_sample,
|
||||||
draw_pred=False,
|
draw_pred=False,
|
||||||
show=not args.not_show,
|
show=not args.not_show,
|
||||||
wait_time=args.show_interval,
|
wait_time=args.show_interval,
|
||||||
|
|
Loading…
Reference in New Issue