DILHTWD
/

documentlayoutsegmentation_YOLOv8_ondoclaynet

Model card Files Files and versions

DILHTWD commited on Mar 17, 2024

Commit

510a020

·

verified ·

1 Parent(s): 2e320b2

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -40,7 +40,7 @@ import pathlib
 img_list = ['sample1.png', 'sample2.png', 'sample3.png']
 # Load the document segmentation model
-docseg_model = YOLO('yolov8x-doclaynetcore-imgsz640-lr1.0000e-02.pt')
 # Process the images with the model
 results = docseg_model(source=img_list, save=True, show_labels=True, show_conf=True, show_boxes=True)

 img_list = ['sample1.png', 'sample2.png', 'sample3.png']
 # Load the document segmentation model
+docseg_model = YOLO('yolov8x-doclaynet-epoch64-imgsz640-initiallr1e-4-finallr1e-5.pt')
 # Process the images with the model
 results = docseg_model(source=img_list, save=True, show_labels=True, show_conf=True, show_boxes=True)