Browse files


1 parent 3c81602 commit bd8c9d879f957e1218a8f9e1f9b663ac70375866 @karpathy committed Sep 23, 2016
Showing with 2 additions and 0 deletions.
  1. +2 −0
@@ -1,6 +1,8 @@
# NeuralTalk2
+**Update (September 22, 2016)**: The Google Brain team has [released the image captioning model]( of Vinyals et al. (2015). The core model is very similar to NeuralTalk2 (a CNN followed by RNN), but the Google release should work significantly better as a result of better CNN, some tricks, and more careful engineering. Find it under [im2txt]( repo in tensorflow. I'll leave this code base up for educational purposes and as a Torch implementation.
Recurrent Neural Network captions your images. Now much faster and better than the original [NeuralTalk]( Compared to the original NeuralTalk this implementation is **batched, uses Torch, runs on a GPU, and supports CNN finetuning**. All of these together result in quite a large increase in training speed for the Language Model (~100x), but overall not as much because we also have to forward a VGGNet. However, overall very good models can be trained in 2-3 days, and they show a much better performance.
This is an early code release that works great but is slightly hastily released and probably requires some code reading of inline comments (which I tried to be quite good with in general). I will be improving it over time but wanted to push the code out there because I promised it to too many people.

0 comments on commit bd8c9d8

Please sign in to comment.