Hugging Face's logo

DILHTWD
/
documentlayoutsegmentation_YOLOv8_ondoclaynet

Model card Files Community
2
DILHTWD commited on
Commit
510a020
·
verified ·
1 Parent(s): 2e320b2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -40,7 +40,7 @@ import pathlib
40
  img_list = ['sample1.png', 'sample2.png', 'sample3.png']
41
 
42
  # Load the document segmentation model
43
- docseg_model = YOLO('yolov8x-doclaynetcore-imgsz640-lr1.0000e-02.pt')
44
 
45
  # Process the images with the model
46
  results = docseg_model(source=img_list, save=True, show_labels=True, show_conf=True, show_boxes=True)
 
40
  img_list = ['sample1.png', 'sample2.png', 'sample3.png']
41
 
42
  # Load the document segmentation model
43
+ docseg_model = YOLO('yolov8x-doclaynet-epoch64-imgsz640-initiallr1e-4-finallr1e-5.pt')
44
 
45
  # Process the images with the model
46
  results = docseg_model(source=img_list, save=True, show_labels=True, show_conf=True, show_boxes=True)