In [1]:
import theano
import os, sys
sys.path.insert(1, os.path.join('utils'))
path = 'data/statefarm/'
import utils; reload(utils)
from utils import *


/home/wnixalo/miniconda3/envs/FAI/lib/python2.7/site-packages/theano/gpuarray/dnn.py:135: UserWarning: Your cuDNN version is more recent than Theano. If you encounter problems, try updating Theano or downgrading cuDNN to version 5.1.
  warnings.warn("Your cuDNN version is more recent than "
Using cuDNN version 6021 on context None
Mapped name None to device cuda: GeForce GTX 870M (0000:01:00.0)
Using Theano backend.

In [2]:
batch_size=16
vgg = Vgg16()
model = vgg.model
last_conv_idx = [i for i, l in enumerate(model.layers) if type(l) is Convolution2D][-1]
conv_layers = model.layers[:last_conv_idx + 1]
conv_model = Sequential(conv_layers)

In [5]:
test_batches = get_batches(path + 'test', batch_size=batch_size, shuffle=False)


Found 12500 images belonging to 1 classes.

In [4]:
path = 'data/catsdogs_rdx/'

In [6]:
path = 'data/catsdogs_rdx'
test_batches = get_batches(path + '/test', batch_size=batch_size, shuffle=False)


Found 12500 images belonging to 1 classes.

In [ ]: