An open API service indexing awesome lists of open source software.

https://github.com/facebookarchive/itorch

IPython kernel for Torch with visualization and plotting
https://github.com/facebookarchive/itorch

Last synced: 4 months ago
JSON representation

IPython kernel for Torch with visualization and plotting

Awesome Lists containing this project

README

          

# iTorch
iTorch is an [IPython](http://ipython.org) Kernel for Torch, with plotting (using [Bokeh.js](http://bokeh.pydata.org/docs/gallery.html) plots) and visualization of images, video and audio

## Features
iTorch in notebook mode works like any other IPython notebook.
It provides useful **inline auto-complete**. Whenever you need auto-complete, use the **TAB key**.
![screenshot](screenshots/autocomplete.png "")

It also provides **inline help** using the ? symbol.
For example, `?torch.cmul`
![screenshot](screenshots/help.png "")

In addition, we introduce visualization functions for images, video, audio, html and plots.

**itorch.image(img)** - You can pass in a 3-D tensor (for a single image), or a table of 3D tensors (image collages).
```lua
itorch.image({image.lena(), image.lena(), image.lena()})
```
![screenshot](screenshots/image.png "")

**itorch.audio(path)** - You can pass in a filename of an audio file. Formats supported are mp3, ogg, aac
```lua
itorch.audio('example.mp3')
```
![screenshot](screenshots/audio.png "")

**itorch.video(path)** - You can pass in a filename of a video file. Formats supported are mp4, ogv, mpeg
```lua
itorch.video('example.mp4')
```
![screenshot](screenshots/video.png "")

**[window-id] = itorch.html(htmlstring, [window-id])** - Raw HTML string that is passed is rendered. A window handle is returned, that can be reused to replace the HTML with something else.
```lua
itorch.html('

Hi there! this is arbitrary HTML

')
window_id = itorch.html('

This text will be replaced in 2 seconds

')
os.execute('sleep 2')
itorch.html('

magic!

', window_id)
```
![screenshot](screenshots/html.png "")

###Plotting
iTorch can plot to screen in notebook mode, or save the plot to disk as a html file.

A Plot object is introduced, that can plot different kinds of plots such as scatter, line, segment, quiver plots.
```lua
Plot = require 'itorch.Plot'
```
The plotting can be extended to more kinds of plots, as it uses [Bokeh.js](http://bokeh.pydata.org/en/latest/docs/dev_guide/bokehjs.html) as its backend.
```lua
x1 = torch.randn(40):mul(100)
y1 = torch.randn(40):mul(100)
x2 = torch.randn(40):mul(100)
y2 = torch.randn(40):mul(100)
x3 = torch.randn(40):mul(200)
y3 = torch.randn(40):mul(200)

-- scatter plots
plot = Plot():circle(x1, y1, 'red', 'hi'):circle(x2, y2, 'blue', 'bye'):draw()
plot:circle(x3,y3,'green', 'yolo'):redraw()
plot:title('Scatter Plot Demo'):redraw()
plot:xaxis('length'):yaxis('width'):redraw()
plot:legend(true)
plot:redraw()
-- print(plot:toHTML())
plot:save('out.html')
```
![screenshot](screenshots/scatter.png "")

```lua
-- line plots
plot = Plot():line(x1, y1,'red','example'):legend(true):title('Line Plot Demo'):draw()
```

```lua
-- segment plots
plot = Plot():segment(x1, y1, x1+10,y1+10, 'red','demo'):title('Segment Plot Demo'):draw()
```

```lua
-- quiver plots
U = torch.randn(3,3):mul(100)
V = torch.randn(3,3):mul(100)
plot = Plot():quiver(U,V,'red',''):title('Quiver Plot Demo'):draw()
```
![screenshot](screenshots/quiver.png "")

```lua
-- quads/rectangles
x1=torch.randn(10)
y1=torch.randn(10)
plot = Plot():quad(x1,y1,x1+1,y1+1,'red',''):draw()
```

```lua
-- histogram
plot = Plot():histogram(torch.randn(10000)):draw()
```
![screenshot](screenshots/hist.png "")

#### Hover Tool in Plotting
```lua
local t = torch.Tensor
local y = t(10)
local x = t(y:size()):zero()
local labels = {}
for i = 1, 10 do
y[i] = i
labels[i] = tostring(i)
end

itorch.Plot()
:circle(x, y, 'red', nil, {foo=labels})
:hover_tool({{'xy', '@x @y'}, {'foo', '@foo'}})
:draw()
```

#### Text method to plotting
```lua
local t = torch.Tensor
local y = t(10)
local x = t(y:size()):zero()
local labels = {}
for i = 1, 10 do
y[i] = i
labels[i] = tostring(i)
end

itorch.Plot():gscatter(x, y)
:text(x, y, labels, y, 'black')
:triangle(x, y, 'blue')
:draw()
```

#### Group Scatter plot
Run the following in itorch to produce plots:
```lua
x = torch.randn(200); y = torch.randn(200); x:narrow(1, 1, 100):add(2);
labels = torch.LongTensor(200):zero(); labels:add(1); labels:narrow(1, 1, 100):add(1)
itorch.Plot():gscatter(x, y):title('Scatter plot without labels'):draw()
itorch.Plot():gscatter(x, y, labels):title('Scatter plot with labels and legend #1'):legend(true):draw()
itorch.Plot():gscatter(x, y, labels, true):title('Scatter plot with labels and legend #2'):draw()
itorch.Plot():gscatter(x, y, labels, false):title('Scatter plot with labels and no legend'):draw()
```

## Requirements
iTorch requires or works with
* Mac OS X or Linux (tested in Ubuntu 14.04 and Arch Linux)
* [Torch-7](https://github.com/torch/torch7/wiki/Cheatsheet#installing-and-running-torch)
* [IPython](http://ipython.org/install.html) version 2.2 or above (you can check your version of ipython using ipython --version)
* ZeroMQ
```bash
# OSX
brew install zeromq
brew install openssl
luarocks install luacrypto OPENSSL_DIR=/usr/local/opt/openssl/

# Ubuntu
sudo apt-get install libzmq3-dev libssl-dev python-zmq

# Ubuntu 16
luarocks install lzmq
```

## Installing iTorch
```bash
git clone https://github.com/facebook/iTorch.git
cd iTorch
luarocks make
```

If you have to use sudo for some reason (if you globally installed torch on Linux for example), use these commands:
```bash
sudo env "PATH=$PATH" luarocks make
sudo chown -R $USER $(dirname $(ipython locate profile))
```

## Docker Images
Ubuntu 14.04 + iTorch notebook + Miniconda: see docker hub [repo](https://hub.docker.com/r/dhunter/itorch-notebook/).

```bash
docker pull dhunter/itorch-notebook
```

## How iTorch works
Start iTorch at command-line using the following command:
```bash
itorch notebook # notebook mode
OR
itorch # console mode
OR
itorch qtconsole # Qt mode
```

In notebook mode, you can use the "New" button to create a new notebook.

## Examples
Demo iTorch notebook: http://nbviewer.ipython.org/github/facebook/iTorch/blob/master/iTorch_Demo.ipynb

## Join the Torch community
See the CONTRIBUTING file for how to help out.

## License
iTorch is BSD-licensed. We also provide an additional patent grant.