opendatalab / DocLayout-YOLO

DocLayout-YOLO: Enhancing Document Layout Analysis through Diverse Synthetic Data and Global-to-Local Adaptive Perception
https://huggingface.co/spaces/opendatalab/DocLayout-YOLO
GNU Affero General Public License v3.0
141 stars 8 forks source link

about predict_single.py #4

Closed lvpeng159 closed 4 days ago

lvpeng159 commented 4 days ago
ori_code: 

det_res = model.predict( args.image_file, imgsz=args.imgsz, conf=args.conf, device=args.device, ) according to parser.add_argument, it should be: det_res = model.predict( args.image_path, # args.image_file imgsz=args.imgsz, conf=args.conf, device=args.device, )

JulioZhao97 commented 4 days ago

Hello, thanks for your question. The inference script is now demo.py in main branch.