Skip to content

Commit

Permalink
Add weights path
Browse files Browse the repository at this point in the history
  • Loading branch information
tomassosorio committed Feb 11, 2021
1 parent a331ded commit 5f4d781
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,12 @@ To train is only needed the `train.py` file which can be configured as wanted.
To predict, it can be used the pre-trained weights already available and should be downloaded on the following link: [TableNet Weights](https://drive.google.com/drive/folders/1YbdQQ3ZLjrltfu7yBm7G5uVt2RYkWLoM?usp=sharing)

```bash
python predict.py --image_path=<image path>'
python predict.py --model_weights='<weights path>' --image_path='<image path>'
```

or simply:
```bash
python predict.py'
python predict.py
```

To predict with the default image.

0 comments on commit 5f4d781

Please sign in to comment.