You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
-`data_dir`: Directory containing the data h5 files. Default is `Data/`.
37
37
38
38
- <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"```
40
40
* 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.
41
41
42
42
- <b>Evaluation</b>
@@ -55,7 +55,7 @@ The model achieved an accuray of 50.8% on the validation dataset after 12 epochs
55
55
56
56
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:
57
57
```
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"
59
59
```
60
60
| Image | Question | Top Answers (left to right) |
0 commit comments