diff --git a/tutorials/03-advanced/image_captioning/README.md b/tutorials/03-advanced/image_captioning/README.md index 5346ac7..7a658a2 100644 --- a/tutorials/03-advanced/image_captioning/README.md +++ b/tutorials/03-advanced/image_captioning/README.md @@ -51,4 +51,9 @@ $ python train.py ```bash $ python sample.py --image='png/example.png' -``` \ No newline at end of file +``` + +
+ +## Pretrained model +If you do not want to train the model from scratch, 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/bmo30z81a4v7m0r/pretrained_model.zip?dl=0) and extract it to `./models/` directory using `unzip pretrained_model.zip`.