Update README.md

This commit is contained in:
Baek JeongHun 2020-02-25 20:27:31 +09:00 committed by GitHub
parent ffc3e01ad5
commit 8e7aba3b5d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -22,7 +22,8 @@ The difference between our paper and ICDAR challenge is summarized [here](https:
## Getting Started
### Dependency
- This work was tested with PyTorch 1.3.1, CUDA 9.0, python 3.6 and Ubuntu 16.04. <br> You may need `pip3 install torch==1.3.1`. In the paper, expriments were performed with **PyTorch 0.4.1**.
- This work was tested with PyTorch 1.3.1, CUDA 10.1, python 3.6 and Ubuntu 16.04. <br> You may need `pip3 install torch==1.3.1`. <br>
In the paper, expriments were performed with **PyTorch 0.4.1, CUDA 9.0**.
- requirements : lmdb, pillow, torchvision, nltk, natsort
```
pip3 install lmdb pillow torchvision nltk natsort