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 *
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)
In [4]:
path = 'data/catsdogs_rdx/'
In [6]:
path = 'data/catsdogs_rdx'
test_batches = get_batches(path + '/test', batch_size=batch_size, shuffle=False)
In [ ]: