Netra-OCR / layout /netra-model.json
Darayut's picture
Store the recognizers' and layout model's weights in fp16 (half the download)
04cb751 verified
Raw History Blame Contribute Delete
1.09 kB
{
"format": "netra-bundle",
"format_version": 1,
"name": "netra-ocr-layout",
"version": "1.0.0",
"task": "object-detection",
"graphs": {
"model": "model.onnx"
},
"components": {
"preprocess": {
"type": "letterbox",
"imgsz": 1024,
"stride": 32,
"auto": true,
"pad_value": 114
},
"head": {
"type": "yolo-end2end",
"labels": [
"title",
"plain text",
"abandon",
"figure",
"figure_caption",
"table",
"table_caption",
"table_footnote",
"isolate_formula",
"formula_caption"
],
"conf": 0.2
}
},
"metadata": {
"license": "apache-2.0",
"source_checkpoint": "doclayout.pt"
},
"variants": {
"fp16": {
"graphs": {
"model": "model.fp16.onnx"
},
"devices": [
"cuda"
]
}
},
"files": {
"model.fp16.onnx": "6fbbe1a2d71b09f937e50b46b81adad0a54f06fcbae406a0fa3c2aafaa1585f2",
"model.onnx": "87b1b6b983f7120547342ab23d0564fb2b3766f23048a297590d73919aee8383"
}
}