mirror of https://github.com/open-mmlab/mmocr.git
33 lines
1.1 KiB
Markdown
33 lines
1.1 KiB
Markdown
|
# 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
|
||
|
|
```
|