Skip to content

Commit 6c0cc8c

Browse files
Update README.md
1 parent f103398 commit 6c0cc8c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ python extract_fc7.py --split=val
3636
- `data_dir`: Directory containing the data h5 files. Default is `Data/`.
3737

3838
- <b>Prediction</b>
39-
* ```python predict.py --image_path="sample_image.jpg" --question="What is the color of the animal shown?" --model_path = "Data/Models/model25.ckpt"```
39+
* ```python predict.py --image_path="sample_image.jpg" --question="What is the color of the animal shown?" --model_path = "Data/Models/model2.ckpt"```
4040
* Models are saved during training after each of the complete training data in ```Data/Models```. Supply the path of the trained model in ```model_path``` option.
4141

4242
- <b>Evaluation</b>
@@ -55,7 +55,7 @@ The model achieved an accuray of 50.8% on the validation dataset after 12 epochs
5555

5656
The fun part! Try it for yourself. Make sure you have tensorflow installed. Download the data files/trained model from [this link][9] and save them in the ```Data/``` directory. Also download the [pretrained VGG-16 model][7] and save it as ```Data/vgg16.tfmodel```. You can test for any sample image using:
5757
```
58-
python predict.py --image_path="Data/sample.jpg" --question="Which animal is this?" --model_path="Data/model7.ckpt"
58+
python predict.py --image_path="Data/sample.jpg" --question="Which animal is this?" --model_path="Data/model2.ckpt"
5959
```
6060
| Image | Question | Top Answers (left to right) |
6161
| ------------- |:-------------:| -----:|

0 commit comments

Comments
 (0)