search

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
533 stars 38 forks source link

about label.yaml #7

Closed lvpeng159 closed 1 month ago

lvpeng159 commented 1 month ago

i use the “doclayout_yolo_docstructbench_imgsz1024.pt” weight file，does this project have a cls.yaml file for this weight file to show all the classes? maybe it exists，i just can‘t find it. ：)

JulioZhao97 commented 1 month ago

The classes to id mapping is as follows:

id_to_names = {
    0: 'title', 
    1: 'plain text',
    2: 'abandon', 
    3: 'figure', 
    4: 'figure_caption', 
    5: 'table', 
    6: 'table_caption', 
    7: 'table_footnote', 
    8: 'isolate_formula', 
    9: 'formula_caption'
}
JulioZhao97 commented 1 month ago

If there are any other questions, feel free to let me know!

lvpeng159 commented 1 month ago

If there are any other questions, feel free to let me know!

ok，Thank you for your reply response.：）