Layout-Parser / layout-parser

A Unified Toolkit for Deep Learning Based Document Image Analysis
https://layout-parser.github.io/
Apache License 2.0
4.75k stars 456 forks source link

Save the model.pth #63

Closed maiia-bocharova closed 2 years ago

maiia-bocharova commented 2 years ago

Hello, how can I download the model.pth for further use? I am able to download the config from the model_zoo, but where do I find the final_model.pth file?

Upd: I have found the files in src/layoutparser/models/detectron2/catalog.py Sorry for asking this

lolipopshock commented 2 years ago

In addition to this, if you have already have the models downloaded in layoutparser, you can also go to ~/.torch/iopath_cache/s to find the downloaded files -- you might need to refer to the catalog.py to figure out the model names based on the.