Spaces:
Runtime error
Runtime error
# Useful Tools | |
We provide some useful tools under `mmocr/tools` directory. | |
## Publish a Model | |
Before you upload a model to AWS, you may want to | |
(1) convert the model weights to CPU tensors, (2) delete the optimizer states and | |
(3) compute the hash of the checkpoint file and append the hash id to the filename. These functionalities could be achieved by `tools/publish_model.py`. | |
```shell | |
python tools/publish_model.py ${INPUT_FILENAME} ${OUTPUT_FILENAME} | |
``` | |
For example, | |
```shell | |
python tools/publish_model.py work_dirs/psenet/latest.pth psenet_r50_fpnf_sbn_1x_20190801.pth | |
``` | |
The final output filename will be `psenet_r50_fpnf_sbn_1x_20190801-{hash id}.pth`. | |
## Convert txt annotation to lmdb format | |
Sometimes, loading a large txt annotation file with multiple workers can cause OOM (out of memory) error. You can convert the file into lmdb format using `tools/data/utils/txt2lmdb.py` and use LmdbLoader in your config to avoid this issue. | |
```bash | |
python tools/data/utils/txt2lmdb.py -i <txt_label_path> -o <lmdb_label_path> | |
``` | |
For example, | |
```bash | |
python tools/data/utils/txt2lmdb.py -i data/mixture/Syn90k/label.txt -o data/mixture/Syn90k/label.lmdb | |
``` | |