Mirrors/dots.ocr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

dots.ocr-1.5 dots.ocr-1.5-svg released

Browse Source
This commit is contained in:
zhangwei13
2026-02-16 01:53:00 +08:00
parent c3084bd437
commit d2e359c715
14 changed files with 1407 additions and 981 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/download_model.py
+2 -2
View File
@@ -5,11 +5,11 @@ import os
if __name__ == '__main__':
parser = ArgumentParser()
parser.add_argument('--type', '-t', type=str, default="huggingface")
parser.add_argument('--name', '-n', type=str, default="rednote-hilab/dots.ocr")
parser.add_argument('--name', '-n', type=str, default="rednote-hilab/dots.ocr-1.5")
args = parser.parse_args()
script_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
print(f"Attention: The model save dir dots.ocr should be replace by a name without `.` like DotsOCR, util we merge our code to transformers.")
model_dir = os.path.join(script_dir, "weights/DotsOCR")
model_dir = os.path.join(script_dir, "weights/DotsOCR_1_5")
if not os.path.exists(model_dir):
os.makedirs(model_dir)
if args.type == "huggingface":