Skip to content

Moodstocks/inception-v3.torch

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

Torch port of Inception V3

Scripts to dump TensorFlow Inception V3 weights and to reconstruct the network in Torch.

The approach is inspired by soumith/inception.torch.

Overview

  • dump_filters.py: a Python/TensorFlow script to dump all the weights of Inception V3
  • inceptionv3.lua: reads the weights and builds the Torch binary equivalent network
  • example.lua: example use of the Torch network

Usage

Step 1: TensorFlow

Here are instructions using Docker:

# From the host
docker run -it \
-p 8888:8888 \
-v /home/myuser/code/inception-v3.torch/dump_filters.py:/root/dump_filters.py \
-v /home/myuser/data/dump:/root/dump \
gcr.io/tensorflow/tensorflow

# From the container
apt-get update
apt-get install -y libhdf5-dev
pip install h5py
python dump_filters.py

If you have already installed TensorFlow, just run dump_filters.py and the script will generate a directory dump with all the filters.

Step 2: Torch

Install pre-requisite:

luarocks install hdf5

Given that the filters are dumped in /home/myuser/data/dump, execute:

luajit inceptionv3.lua -i /home/myuser/data/dump \
-o /home/myuser/networks/inceptionv3.net
-b cudnn

The parameter -b sets the backend to use: nn, cunn, or cudnn. The produced binary Torch model will be saved in /home/myuser/networks/inceptionv3.net.

Test it with an image as follows:

luajit example.lua -m /home/myuser/networks/inceptionv3.net \
-b cudnn \
-i myimage.jpg \
-s synsets.txt

With TensorFlow example image you should obtain a result like this:

RESULTS (top-5):
----------------
score = 0.847576: n02510455 giant panda, panda, panda bear, coon bear, Ailuropoda melanoleuca (170)
score = 0.020494: n02500267 indri, indris, Indri indri, Indri brevicaudatus (76)
score = 0.003694: n02509815 lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens (8)
score = 0.001323: n13044778 earthstar (879)
score = 0.001301: n07760859 custard apple (326)

About

Rethinking the Inception Architecture for Computer Vision

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published