Releasing image recognition for TensorFlow

2,268 views
Skip to first unread message

Jon Shlens

unread,
Dec 7, 2015, 9:02:28 AM12/7/15
to dis...@tensorflow.org

Dear TensorFlow community,


Today we are releasing our best image classifier trained on ImageNet data. As described in our recent Arxiv preprint at http://arxiv.org/abs/1512.00567, an ensemble of four of these models achieves 3.46% top-5 error on the validation set of the ImageNet whole image ILSVRC2012 classification task (compared with our ensemble from last year that won the 2014 ImageNet classification challenge with a 6.66% top-5 error rate).


In this release, we are supplying code and data files containing the trained model parameters for running the image classifier on:

  • Both desktop and mobile environments

  • Employing either a C++ or Python API.


In addition, we are providing a tutorial that describes how to use the image recognition system for a variety of use-cases.

   http://www.tensorflow.org/tutorials/image_recognition/index.html


This release allows one to compute higher-level visual features and/or perform image recognition on the ImageNet 1000 object label set*. We are actively working on refactoring our code so that we can open-source a complete training system for this model so that others can train on their own data or fine-tune it for other purposes (Some custom training operations for this model rely on code that is shared between our non-open-sourced DistBelief system and TensorFlow and we are actively working on disentangling this).


We wish to especially acknowledge Christian Szegedy, Sergey Ioffe and Vincent Vanhoucke for developing this network and helping make this happen.


Happy Holidays-

Sherry, Pete, Chris, Jon with contributions from many team members


* The 1000 object labels are listed at:

http://image-net.org/challenges/LSVRC/2014/browse-synsets


hie...@sixthgearstudios.com

unread,
Dec 7, 2015, 10:26:36 PM12/7/15
to Discuss
Thanks a lot. We're looking forward to future releases. ^_^

jbo...@googlemail.com

unread,
Dec 8, 2015, 1:15:04 PM12/8/15
to Discuss
Awesome. Thanks!
When will this be available in the pip release? The current one does not contain the new classifiers.

Vijay Vasudevan

unread,
Dec 8, 2015, 1:16:25 PM12/8/15
to jbo...@googlemail.com, Discuss
If you git clone the repository, you should be able to run `python classify_image.py` even with the 0.5.0 pip package.  Is there something else you needed?

--
You received this message because you are subscribed to the Google Groups "Discuss" group.
To unsubscribe from this group and stop receiving emails from it, send an email to discuss+u...@tensorflow.org.
To post to this group, send email to dis...@tensorflow.org.
To view this discussion on the web visit https://groups.google.com/a/tensorflow.org/d/msgid/discuss/7de8151e-5232-4cd1-80db-a6b49683fbfb%40tensorflow.org.

sam...@gmail.com

unread,
Jan 26, 2016, 4:10:51 AM1/26/16
to Discuss
Thanks for sharing this! 

Do you have any timeline for the release of the tutorial for the training part? I am eager to start training models on my own datasets :)

Looking forward to the the next release
++

hakan....@gmail.com

unread,
Feb 9, 2016, 8:15:03 PM2/9/16
to Discuss
Hello,
I am also looking for the training scripts so that I can fine-tune it for a different dataset. Are you planning to release scripts for training part also?
Best regards
Hakan

Jon Shlens

unread,
Mar 9, 2016, 5:36:19 PM3/9/16
to hakan....@gmail.com, Discuss
Hi everyone-

To close this thread, please take a look at our blog announcement today.


best-
 Jon


--
You received this message because you are subscribed to the Google Groups "Discuss" group.
To unsubscribe from this group and stop receiving emails from it, send an email to discuss+u...@tensorflow.org.
To post to this group, send email to dis...@tensorflow.org.

ClessVNA

unread,
Mar 14, 2016, 12:48:47 AM3/14/16
to Discuss
Hi, 

I got this error while following the instructions for finetuning on flower dataset:

 Unable to open table file inception-v3-model/model.ckpt-157585: Data loss: corrupted compressed block contents: perhaps your file is in a different file format and you need to use a different restore operator?

Geoffrey Irving

unread,
Mar 14, 2016, 12:56:21 PM3/14/16
to ClessVNA, Discuss
Please report issues with inception at https://github.com/tensorflow/models/issues or https://github.com/tensorflow/tensorflow/issues (depending on whether you think it's inception specific), not in the announcement thread.

Jon Shlens

unread,
Mar 15, 2016, 2:01:24 AM3/15/16
to ClessVNA, Discuss
FYI, please refer to this github issue:

We updated the tar-ball containing the model for download. Would you be able re-download and verify that it works on your end? If you have any updates, please add to the github issue. Thanks.

Reply all
Reply to author
Forward
0 new messages