algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-05 00:24:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
yunjey
2017-04-13 20:12:18 +09:00
committed by GitHub
parent ef803c71de
commit 4098e346de
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tutorials/09 - Image Captioning/README.md
View File

@ -4,6 +4,9 @@
#### 1. Clone the repositories
```bash
$ git clone https://github.com/pdollar/coco.git
$ cd coco/PythonAPI/
$ make
$ cd ../../
$ git clone https://github.com/yunjey/pytorch-tutorial.git
$ cd pytorch-tutorial/tutorials/09 - Image Captioning
```
@ -19,7 +22,8 @@ $ ./donwload.sh
#### 3. Preprocessing
```bash
$ python vocab.py
$ python build_vocab.py
$ python resize.py
```
#### 4. Train the model
@ -32,11 +36,11 @@ $ python train.py
```bash
$ python sample.py --image=sample_image.jpg
$ python sample.py --image='path_for_image'
```
<br>
## Pretrained model
If you do not want to train the model yourself, you can use a pretrained model. I have provided the pretrained model as a zip file. You can download the file [here](https://www.dropbox.com/s/cngzozkk73imjdh/trained_model.zip?dl=0) and extract it to `model` directory.
If you do not want to train the model yourself, you can use a pretrained model. I have provided the pretrained model as a zip file. You can download the file [here](https://www.dropbox.com/s/btubjo73dc953zo/trained_model.egg?dl=0) and extract it to `./models/` directory.