class CNN3_Model(ModelBase):
u"""see: http://aidiary.hatenablog.com/entry/20151007/1444223445"""
def __init__(self, input_size=32):
super(CNN3_Model, self).__init__()
# F.Convolution2D(in_channel, out_channel, filter_size)
self.model = FunctionSet( # 1*32*32 -(conv)-> 20*28*28 -(pool)-> 20*14*14
conv1=F.Convolution2D(1, 20, 5),
# 20*14*14 -(conv)-> 50*10*10 -(pool)-> 50*5*5=1250
conv2=F.Convolution2D(20, 50, 5),
l1=F.Linear(1250, 300),
l2=F.Linear(300, 2))
def forward(self, x_data, y_data, train=True):
u"""return loss, accuracy"""
x, t = Variable(x_data), Variable(y_data)
h1 = F.max_pooling_2d(F.relu(self.model.conv1(x)), 2)
h2 = F.max_pooling_2d(F.relu(self.model.conv2(h1)), 2)
h3 = F.dropout(F.relu(self.model.l1(h2)), train=train)
y = self.model.l2(h3)
# 多クラス分類なので誤差関数としてソフトマックス関数の
# 交差エントロピー関数を用いて、誤差を導出。最低でもlossは必要
return {
"loss": F.softmax_cross_entropy(y, t),
"accuracy": F.accuracy(y, t)
}
class CNN3_Model(ModelBase):
u"""see: http://aidiary.hatenablog.com/entry/20151007/1444223445"""
def __init__(self, input_size=32):
super(CNN3_Model, self).__init__()
# F.Convolution2D(in_channel, out_channel, filter_size)
self.model = FunctionSet( # 1*32*32 -(conv)-> 20*28*28 -(pool)-> 20*14*14
conv1=F.Convolution2D(1, 20, 5),
# 20*14*14 -(conv)-> 50*10*10 -(pool)-> 50*5*5=1250
conv2=F.Convolution2D(20, 50, 5),
l1=F.Linear(1250, 300),
l2=F.Linear(300, 2))
def forward(self, x_data, y_data, train=True):
u"""return loss, accuracy"""
x, t = Variable(x_data), Variable(y_data)
h1 = F.max_pooling_2d(F.relu(self.model.conv1(x)), 2)
h2 = F.max_pooling_2d(F.relu(self.model.conv2(h1)), 2)
h3 = F.dropout(F.relu(self.model.l1(h2)), train=train)
y = self.model.l2(h3)
# 多クラス分類なので誤差関数としてソフトマックス関数の
# 交差エントロピー関数を用いて、誤差を導出。最低でもlossは必要
return {
"loss": F.softmax_cross_entropy(y, t),
"accuracy": F.accuracy(y, t)
}
class DQN_CNN(object):
def __init__(self,n_act):
N_output = n_act
self.model = FunctionSet(
conv1=F.Convolution2D(1, 16, 3, pad=1),
conv2=F.Convolution2D(16, 16, 3, pad=1),
l1=F.Linear(256, 256),
l2=F.Linear(256, N_output))
def Q_func(self,x):
N,h,w=x.shape
x=x.reshape(N,1,h,w)
x = Variable(x)
h = F.relu(self.model.conv1(x))
h = F.max_pooling_2d(F.relu(self.model.conv2(h)), 2)
h = F.relu(self.model.l1(h))
y = self.model.l2(h)
return y
class Inception(Function):
"""Inception module of GoogLeNet.
It applies four different functions to the input array and concatenates
their outputs along the channel dimension. Three of them are 2D convolutions
of sizes 1x1, 3x3 and 5x5. Convolution paths of 3x3 and 5x5 sizes have 1x1
convolutions (called projections) ahead of them. The other path consists of
1x1 convolution (projection) and 3x3 max pooling.
The output array has the same spatial size as the input. In order to satisfy
this, Inception module uses appropriate padding for each convolution and
pooling.
See: `Going Deeper with Convolutions <http://arxiv.org/abs/1409.4842>`_.
Args:
in_channels (int): Number of channels of input arrays.
out1 (int): Output size of 1x1 convolution path.
proj3 (int): Projection size of 3x3 convolution path.
out3 (int): Output size of 3x3 convolution path.
proj5 (int): Projection size of 5x5 convolution path.
out5 (int): Output size of 5x5 convolution path.
proj_pool (int): Projection size of max pooling path.
Returns:
Variable: Output variable. Its array has the same spatial size and the
same minibatch size as the input array. The channel dimension has
size ``out1 + out3 + out5 + proj_pool``.
.. note::
This function inserts the full computation graph of the Inception module behind
the input array. This function itself is not inserted into the
computation graph.
"""
def __init__(self, in_channels, out1, proj3, out3, proj5, out5, proj_pool):
self.f = FunctionSet(
conv1 = Convolution2D(in_channels, out1, 1),
proj3 = Convolution2D(in_channels, proj3, 1),
conv3 = Convolution2D(proj3, out3, 3, pad=1),
proj5 = Convolution2D(in_channels, proj5, 1),
conv5 = Convolution2D(proj5, out5, 5, pad=2),
projp = Convolution2D(in_channels, proj_pool, 1),
)
def forward(self, x):
self.x = Variable(x[0])
out1 = self.f.conv1(self.x)
out3 = self.f.conv3(relu(self.f.proj3(self.x)))
out5 = self.f.conv5(relu(self.f.proj5(self.x)))
pool = self.f.projp(max_pooling_2d(self.x, 3, stride=1, pad=1))
self.y = relu(concat((out1, out3, out5, pool), axis=1))
return self.y.data,
def backward(self, x, gy):
self.y.grad = gy[0]
self.y.backward()
return self.x.grad,
def to_gpu(self, device=None):
return self.f.to_gpu(device)
def to_cpu(self):
return self.f.to_cpu()
@property
def parameters(self):
return self.f.parameters
@parameters.setter
def parameters(self, params):
self.f.parameters = params
@property
def gradients(self):
return self.f.gradients
@gradients.setter
def gradients(self, grads):
self.f.gradients = grads
class DQN_class:
# Hyper-Parameters
gamma = 0.99 # Discount factor
def __init__(self, enable_controller=[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17]):
self.num_of_actions = len(enable_controller)
self.enable_controller = enable_controller # Default setting : "Pong"
print "Initializing DQN..."
# Initialization of Chainer 1.1.0 or older.
# print "CUDA init"
# cuda.init()
print "Model Building"
w = math.sqrt(2) # MSRA scaling
self.model = FunctionSet(
conv1=F.Convolution2D(3, 64, 7, wscale=w, stride=2, pad=3),
conv2_1_1=F.Convolution2D(64, 64, 1, wscale=w, stride=1),
conv2_1_2=F.Convolution2D(64, 64, 3, wscale=w, stride=1, pad=1),
conv2_1_3=F.Convolution2D(64, 256, 1, wscale=w, stride=1),
conv2_1_ex=F.Convolution2D(64, 256, 1, wscale=w, stride=1),
conv2_2_1=F.Convolution2D(256, 64, 1, wscale=w, stride=1),
conv2_2_2=F.Convolution2D(64, 64, 3, wscale=w, stride=1, pad=1),
conv2_2_3=F.Convolution2D(64, 256, 1, wscale=w, stride=1),
conv2_3_1=F.Convolution2D(256, 64, 1, wscale=w, stride=1),
conv2_3_2=F.Convolution2D(64, 64, 3, wscale=w, stride=1, pad=1),
conv2_3_3=F.Convolution2D(64, 256, 1, wscale=w, stride=1),
conv3_1_1=F.Convolution2D(256, 128, 1, wscale=w, stride=2),
conv3_1_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_1_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_1_ex=F.Convolution2D(256, 512, 1, wscale=w, stride=2),
conv3_2_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_2_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_2_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_3_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_3_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_3_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_4_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_4_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_4_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_5_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_5_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_5_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_6_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_6_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_6_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_7_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_7_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_7_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv3_8_1=F.Convolution2D(512, 128, 1, wscale=w, stride=1),
conv3_8_2=F.Convolution2D(128, 128, 3, wscale=w, stride=1, pad=1),
conv3_8_3=F.Convolution2D(128, 512, 1, wscale=w, stride=1),
conv4_1_1=F.Convolution2D(512, 256, 1, wscale=w, stride=2),
conv4_1_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_1_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_1_ex=F.Convolution2D(512, 1024, 1, wscale=w, stride=2),
conv4_2_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_2_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_2_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_3_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_3_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_3_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_4_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_4_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_4_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_5_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_5_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_5_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_6_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_6_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_6_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_7_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_7_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_7_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_8_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_8_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_8_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_9_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_9_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_9_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_10_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_10_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_10_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_11_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_11_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_11_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_12_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_12_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_12_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_13_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_13_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_13_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_14_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_14_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_14_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_15_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_15_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_15_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_16_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_16_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_16_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_17_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_17_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_17_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_18_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_18_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_18_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_19_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_19_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_19_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_20_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_20_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_20_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_21_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_21_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_21_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_22_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_22_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_22_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_23_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_23_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_23_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_24_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_24_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_24_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_25_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_25_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_25_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_26_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_26_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_26_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_27_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_27_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_27_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_28_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_28_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_28_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_29_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_29_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_29_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_30_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_30_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_30_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_31_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_31_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_31_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_32_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_32_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_32_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_33_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_33_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_33_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_34_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_34_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_34_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_35_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_35_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_35_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv4_36_1=F.Convolution2D(1024, 256, 1, wscale=w, stride=1),
conv4_36_2=F.Convolution2D(256, 256, 3, wscale=w, stride=1, pad=1),
conv4_36_3=F.Convolution2D(256, 1024, 1, wscale=w, stride=1),
conv5_1_1=F.Convolution2D(1024, 512, 1, wscale=w, stride=2),
conv5_1_2=F.Convolution2D(512, 512, 3, wscale=w, stride=1, pad=1),
conv5_1_3=F.Convolution2D(512, 2048, 1, wscale=w, stride=1),
conv5_1_ex=F.Convolution2D(1024, 2048, 1, wscale=w, stride=2),
conv5_2_1=F.Convolution2D(2048, 512, 1, wscale=w, stride=1),
conv5_2_2=F.Convolution2D(512, 512, 3, wscale=w, stride=1, pad=1),
conv5_2_3=F.Convolution2D(512, 2048, 1, wscale=w, stride=1),
conv5_3_1=F.Convolution2D(2048, 512, 1, wscale=w, stride=1),
conv5_3_2=F.Convolution2D(512, 512, 3, wscale=w, stride=1, pad=1),
conv5_3_3=F.Convolution2D(512, 2048, 1, wscale=w, stride=1),
q_value=F.Linear(2048, self.num_of_actions,
initialW=np.zeros((self.num_of_actions, 2048),
dtype=np.float32))
)
self.model_target = copy.deepcopy(self.model)
print "Initizlizing Optimizer"
self.optimizer = optimizers.Adam()
self.optimizer.setup(self.model.collect_parameters())
def forward(self, state, action, Reward, state_dash, episode_end):
num_of_batch = state.shape[0]
s = Variable(state)
s_dash = Variable(state_dash)
Q = self.Q_func(s) # Get Q-value
# Generate Target Signals
tmp = self.Q_func_target(s_dash) # Q(s',*)
tmp = list(map(np.max, tmp.data.get())) # max_a Q(s',a)
max_Q_dash = np.asanyarray(tmp, dtype=np.float32)
target = np.asanyarray(Q.data.get(), dtype=np.float32)
for i in xrange(num_of_batch):
if not episode_end[i][0]:
tmp_ = np.sign(Reward[i]) + self.gamma * max_Q_dash[i]
else:
tmp_ = np.sign(Reward[i])
action_index = self.action_to_index(action[i])
target[i, action_index] = tmp_
# TD-error clipping
td = Variable(cuda.to_gpu(target)) - Q # TD error
td_tmp = td.data + 1000.0 * (abs(td.data) <= 1) # Avoid zero division
td_clip = td * (abs(td.data) <= 1) + td/abs(td_tmp) * (abs(td.data) > 1)
zero_val = Variable(cuda.to_gpu(np.zeros((num_of_batch, self.num_of_actions), dtype=np.float32)))
loss = F.mean_squared_error(td_clip, zero_val)
return loss, Q
def Q_func(self, state):
h = F.relu(self.model.conv1(state))
h = F.max_pooling_2d(h, 3, stride=2)
h_rem = self.model.conv2_1_ex(h)
h = F.relu(self.model.conv2_1_1(h))
h = F.relu(self.model.conv2_1_2(h))
h = self.model.conv2_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv2_2_1(h))
h = F.relu(self.model.conv2_2_2(h))
h = self.model.conv2_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv2_3_1(h))
h = F.relu(self.model.conv2_3_2(h))
h = self.model.conv2_3_3(h)
h = F.relu(h + h_rem)
h_rem = self.model.conv3_1_ex(h)
h = F.relu(self.model.conv3_1_1(h))
h = F.relu(self.model.conv3_1_2(h))
h = self.model.conv3_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_2_1(h))
h = F.relu(self.model.conv3_2_2(h))
h = self.model.conv3_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_3_1(h))
h = F.relu(self.model.conv3_3_2(h))
h = self.model.conv3_3_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_4_1(h))
h = F.relu(self.model.conv3_4_2(h))
h = self.model.conv3_4_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_5_1(h))
h = F.relu(self.model.conv3_5_2(h))
h = self.model.conv3_5_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_6_1(h))
h = F.relu(self.model.conv3_6_2(h))
h = self.model.conv3_6_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_7_1(h))
h = F.relu(self.model.conv3_7_2(h))
h = self.model.conv3_7_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv3_8_1(h))
h = F.relu(self.model.conv3_8_2(h))
h = self.model.conv3_8_3(h)
h = F.relu(h + h_rem)
h_rem = self.model.conv4_1_ex(h)
h = F.relu(self.model.conv4_1_1(h))
h = F.relu(self.model.conv4_1_2(h))
h = self.model.conv4_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_2_1(h))
h = F.relu(self.model.conv4_2_2(h))
h = self.model.conv4_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_3_1(h))
h = F.relu(self.model.conv4_3_2(h))
h = self.model.conv4_3_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_4_1(h))
h = F.relu(self.model.conv4_4_2(h))
h = self.model.conv4_4_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_5_1(h))
h = F.relu(self.model.conv4_5_2(h))
h = self.model.conv4_5_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_6_1(h))
h = F.relu(self.model.conv4_6_2(h))
h = self.model.conv4_6_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_7_1(h))
h = F.relu(self.model.conv4_7_2(h))
h = self.model.conv4_7_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_8_1(h))
h = F.relu(self.model.conv4_8_2(h))
h = self.model.conv4_8_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_9_1(h))
h = F.relu(self.model.conv4_9_2(h))
h = self.model.conv4_9_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_10_1(h))
h = F.relu(self.model.conv4_10_2(h))
h = self.model.conv4_10_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_11_1(h))
h = F.relu(self.model.conv4_11_2(h))
h = self.model.conv4_11_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_12_1(h))
h = F.relu(self.model.conv4_12_2(h))
h = self.model.conv4_12_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_13_1(h))
h = F.relu(self.model.conv4_13_2(h))
h = self.model.conv4_13_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_14_1(h))
h = F.relu(self.model.conv4_14_2(h))
h = self.model.conv4_14_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_15_1(h))
h = F.relu(self.model.conv4_15_2(h))
h = self.model.conv4_15_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_16_1(h))
h = F.relu(self.model.conv4_16_2(h))
h = self.model.conv4_16_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_17_1(h))
h = F.relu(self.model.conv4_17_2(h))
h = self.model.conv4_17_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_18_1(h))
h = F.relu(self.model.conv4_18_2(h))
h = self.model.conv4_18_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_19_1(h))
h = F.relu(self.model.conv4_19_2(h))
h = self.model.conv4_19_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_20_1(h))
h = F.relu(self.model.conv4_20_2(h))
h = self.model.conv4_20_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_21_1(h))
h = F.relu(self.model.conv4_21_2(h))
h = self.model.conv4_21_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_22_1(h))
h = F.relu(self.model.conv4_22_2(h))
h = self.model.conv4_22_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_23_1(h))
h = F.relu(self.model.conv4_23_2(h))
h = self.model.conv4_23_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_24_1(h))
h = F.relu(self.model.conv4_24_2(h))
h = self.model.conv4_24_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_25_1(h))
h = F.relu(self.model.conv4_25_2(h))
h = self.model.conv4_25_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_26_1(h))
h = F.relu(self.model.conv4_26_2(h))
h = self.model.conv4_26_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_27_1(h))
h = F.relu(self.model.conv4_27_2(h))
h = self.model.conv4_27_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_28_1(h))
h = F.relu(self.model.conv4_28_2(h))
h = self.model.conv4_28_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_29_1(h))
h = F.relu(self.model.conv4_29_2(h))
h = self.model.conv4_29_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_30_1(h))
h = F.relu(self.model.conv4_30_2(h))
h = self.model.conv4_30_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_31_1(h))
h = F.relu(self.model.conv4_31_2(h))
h = self.model.conv4_31_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_32_1(h))
h = F.relu(self.model.conv4_32_2(h))
h = self.model.conv4_32_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_33_1(h))
h = F.relu(self.model.conv4_33_2(h))
h = self.model.conv4_33_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_34_1(h))
h = F.relu(self.model.conv4_34_2(h))
h = self.model.conv4_34_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_35_1(h))
h = F.relu(self.model.conv4_35_2(h))
h = self.model.conv4_35_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv4_36_1(h))
h = F.relu(self.model.conv4_36_2(h))
h = self.model.conv4_36_3(h)
h = F.relu(h + h_rem)
h_rem = self.model.conv5_1_ex(h)
h = F.relu(self.model.conv5_1_1(h))
h = F.relu(self.model.conv5_1_2(h))
h = self.model.conv5_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv5_2_1(h))
h = F.relu(self.model.conv5_2_2(h))
h = self.model.conv5_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model.conv5_3_1(h))
h = F.relu(self.model.conv5_3_2(h))
h = self.model.conv5_3_3(h)
h = F.relu(h + h_rem)
h = F.average_pooling_2d(h, 7)
Q = self.model.q_value(h)
return Q
def Q_func_target(self, state):
h = F.relu(self.model_target.conv1(state))
h = F.max_pooling_2d(h, 3, stride=2)
h_rem = self.model_target.conv2_1_ex(h)
h = F.relu(self.model_target.conv2_1_1(h))
h = F.relu(self.model_target.conv2_1_2(h))
h = self.model_target.conv2_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv2_2_1(h))
h = F.relu(self.model_target.conv2_2_2(h))
h = self.model_target.conv2_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv2_3_1(h))
h = F.relu(self.model_target.conv2_3_2(h))
h = self.model_target.conv2_3_3(h)
h = F.relu(h + h_rem)
h_rem = self.model_target.conv3_1_ex(h)
h = F.relu(self.model_target.conv3_1_1(h))
h = F.relu(self.model_target.conv3_1_2(h))
h = self.model_target.conv3_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_2_1(h))
h = F.relu(self.model_target.conv3_2_2(h))
h = self.model_target.conv3_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_3_1(h))
h = F.relu(self.model_target.conv3_3_2(h))
h = self.model_target.conv3_3_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_4_1(h))
h = F.relu(self.model_target.conv3_4_2(h))
h = self.model_target.conv3_4_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_5_1(h))
h = F.relu(self.model_target.conv3_5_2(h))
h = self.model_target.conv3_5_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_6_1(h))
h = F.relu(self.model_target.conv3_6_2(h))
h = self.model_target.conv3_6_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_7_1(h))
h = F.relu(self.model_target.conv3_7_2(h))
h = self.model_target.conv3_7_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv3_8_1(h))
h = F.relu(self.model_target.conv3_8_2(h))
h = self.model_target.conv3_8_3(h)
h = F.relu(h + h_rem)
h_rem = self.model_target.conv4_1_ex(h)
h = F.relu(self.model_target.conv4_1_1(h))
h = F.relu(self.model_target.conv4_1_2(h))
h = self.model_target.conv4_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_2_1(h))
h = F.relu(self.model_target.conv4_2_2(h))
h = self.model_target.conv4_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_3_1(h))
h = F.relu(self.model_target.conv4_3_2(h))
h = self.model_target.conv4_3_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_4_1(h))
h = F.relu(self.model_target.conv4_4_2(h))
h = self.model_target.conv4_4_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_5_1(h))
h = F.relu(self.model_target.conv4_5_2(h))
h = self.model_target.conv4_5_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_6_1(h))
h = F.relu(self.model_target.conv4_6_2(h))
h = self.model_target.conv4_6_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_7_1(h))
h = F.relu(self.model_target.conv4_7_2(h))
h = self.model_target.conv4_7_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_8_1(h))
h = F.relu(self.model_target.conv4_8_2(h))
h = self.model_target.conv4_8_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_9_1(h))
h = F.relu(self.model_target.conv4_9_2(h))
h = self.model_target.conv4_9_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_10_1(h))
h = F.relu(self.model_target.conv4_10_2(h))
h = self.model_target.conv4_10_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_11_1(h))
h = F.relu(self.model_target.conv4_11_2(h))
h = self.model_target.conv4_11_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_12_1(h))
h = F.relu(self.model_target.conv4_12_2(h))
h = self.model_target.conv4_12_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_13_1(h))
h = F.relu(self.model_target.conv4_13_2(h))
h = self.model_target.conv4_13_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_14_1(h))
h = F.relu(self.model_target.conv4_14_2(h))
h = self.model_target.conv4_14_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_15_1(h))
h = F.relu(self.model_target.conv4_15_2(h))
h = self.model_target.conv4_15_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_16_1(h))
h = F.relu(self.model_target.conv4_16_2(h))
h = self.model_target.conv4_16_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_17_1(h))
h = F.relu(self.model_target.conv4_17_2(h))
h = self.model_target.conv4_17_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_18_1(h))
h = F.relu(self.model_target.conv4_18_2(h))
h = self.model_target.conv4_18_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_19_1(h))
h = F.relu(self.model_target.conv4_19_2(h))
h = self.model_target.conv4_19_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_20_1(h))
h = F.relu(self.model_target.conv4_20_2(h))
h = self.model_target.conv4_20_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_21_1(h))
h = F.relu(self.model_target.conv4_21_2(h))
h = self.model_target.conv4_21_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_22_1(h))
h = F.relu(self.model_target.conv4_22_2(h))
h = self.model_target.conv4_22_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_23_1(h))
h = F.relu(self.model_target.conv4_23_2(h))
h = self.model_target.conv4_23_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_24_1(h))
h = F.relu(self.model_target.conv4_24_2(h))
h = self.model_target.conv4_24_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_25_1(h))
h = F.relu(self.model_target.conv4_25_2(h))
h = self.model_target.conv4_25_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_26_1(h))
h = F.relu(self.model_target.conv4_26_2(h))
h = self.model_target.conv4_26_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_27_1(h))
h = F.relu(self.model_target.conv4_27_2(h))
h = self.model_target.conv4_27_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_28_1(h))
h = F.relu(self.model_target.conv4_28_2(h))
h = self.model_target.conv4_28_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_29_1(h))
h = F.relu(self.model_target.conv4_29_2(h))
h = self.model_target.conv4_29_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_30_1(h))
h = F.relu(self.model_target.conv4_30_2(h))
h = self.model_target.conv4_30_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_31_1(h))
h = F.relu(self.model_target.conv4_31_2(h))
h = self.model_target.conv4_31_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_32_1(h))
h = F.relu(self.model_target.conv4_32_2(h))
h = self.model_target.conv4_32_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_33_1(h))
h = F.relu(self.model_target.conv4_33_2(h))
h = self.model_target.conv4_33_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_34_1(h))
h = F.relu(self.model_target.conv4_34_2(h))
h = self.model_target.conv4_34_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_35_1(h))
h = F.relu(self.model_target.conv4_35_2(h))
h = self.model_target.conv4_35_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv4_36_1(h))
h = F.relu(self.model_target.conv4_36_2(h))
h = self.model_target.conv4_36_3(h)
h = F.relu(h + h_rem)
h_rem = self.model_target.conv5_1_ex(h)
h = F.relu(self.model_target.conv5_1_1(h))
h = F.relu(self.model_target.conv5_1_2(h))
h = self.model_target.conv5_1_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv5_2_1(h))
h = F.relu(self.model_target.conv5_2_2(h))
h = self.model_target.conv5_2_3(h)
h = F.relu(h + h_rem)
h_rem = h
h = F.relu(self.model_target.conv5_3_1(h))
h = F.relu(self.model_target.conv5_3_2(h))
h = self.model_target.conv5_3_3(h)
h = F.relu(h + h_rem)
h = F.average_pooling_2d(h, 7)
Q = self.model_target.q_value(h)
return Q
def e_greedy(self, state, epsilon):
s = Variable(state)
Q = self.Q_func(s)
Q = Q.data
if np.random.rand() < epsilon:
index_action = np.random.randint(0, self.num_of_actions)
print "RANDOM"
else:
index_action = np.argmax(Q.get())
print "GREEDY"
return self.index_to_action(index_action)
def target_model_update(self):
self.model_target = copy.deepcopy(self.model)
def index_to_action(self, index_of_action):
return self.enable_controller[index_of_action]
def action_to_index(self, action):
return self.enable_controller.index(action)
class Inception(Function):
"""Inception module of GoogLeNet.
It applies four different functions to the input array and concatenates
their outputs along the channel dimension. Three of them are 2D convolutions
of sizes 1x1, 3x3 and 5x5. Convolution paths of 3x3 and 5x5 sizes have 1x1
convolutions (called projections) ahead of them. The other path consists of
1x1 convolution (projection) and 3x3 max pooling.
The output array has the same spatial size as the input. In order to satisfy
this, Inception module uses appropriate padding for each convolution and
pooling.
See: `Going Deeper with Convolutions <http://arxiv.org/abs/1409.4842>`_.
Args:
in_channels (int): Number of channels of input arrays.
out1 (int): Output size of 1x1 convolution path.
proj3 (int): Projection size of 3x3 convolution path.
out3 (int): Output size of 3x3 convolution path.
proj5 (int): Projection size of 5x5 convolution path.
out5 (int): Output size of 5x5 convolution path.
proj_pool (int): Projection size of max pooling path.
Returns:
Variable: Output variable. Its array has the same spatial size and the
same minibatch size as the input array. The channel dimension has
size ``out1 + out3 + out5 + proj_pool``.
.. note::
This function inserts the full computation graph of the Inception module behind
the input array. This function itself is not inserted into the
computation graph.
"""
def __init__(self, in_channels, out1, proj3, out3, proj5, out5, proj_pool):
self.f = FunctionSet(
conv1=Convolution2D(in_channels, out1, 1),
proj3=Convolution2D(in_channels, proj3, 1),
conv3=Convolution2D(proj3, out3, 3, pad=1),
proj5=Convolution2D(in_channels, proj5, 1),
conv5=Convolution2D(proj5, out5, 5, pad=2),
projp=Convolution2D(in_channels, proj_pool, 1),
)
def forward(self, x):
self.x = Variable(x[0])
out1 = self.f.conv1(self.x)
out3 = self.f.conv3(relu(self.f.proj3(self.x)))
out5 = self.f.conv5(relu(self.f.proj5(self.x)))
pool = self.f.projp(max_pooling_2d(self.x, 3, stride=1, pad=1))
self.y = relu(concat((out1, out3, out5, pool), axis=1))
return self.y.data,
def backward(self, x, gy):
self.y.grad = gy[0]
self.y.backward()
return self.x.grad,
def to_gpu(self, device=None):
return self.f.to_gpu(device)
def to_cpu(self):
return self.f.to_cpu()
@property
def parameters(self):
return self.f.parameters
@parameters.setter
def parameters(self, params):
self.f.parameters = params
@property
def gradients(self):
return self.f.gradients
@gradients.setter
def gradients(self, grads):
self.f.gradients = grads
class CNN:
file_names = None
def __init__(self):
self.optimizer = optimizers.Adam()
self.model_name = "cnn_nantyara"
if os.path.exists(self.model_name):
self.load_model()
else:
self.crete_model()
self.optimizer.setup(self.model.collect_parameters())
def crete_model(self):
self.model = FunctionSet(
conv1=F.Convolution2D(3, 32, 3),
bn1=F.BatchNormalization(32),
conv2=F.Convolution2D(32, 64, 3, pad=1),
bn2=F.BatchNormalization(64),
conv3=F.Convolution2D(64, 64, 3, pad=1),
fl4=F.Linear(1024, 256),
fl5=F.Linear(256, 2),
)
def get_data(self, ifpath, image_categories, reshape_size=(3, 32, 32)):
x = []
x_apd = x.append
y = []
y_apd = y.append
for i_category, category in enumerate(image_categories):
for i_num in xrange(1, self.get_num_of_images(ifpath, category)):
image = np.array(Image.open(ifpath + "/" + category + str(i_num) + ".jpeg"), dtype=np.float32).reshape(
reshape_size
)
x_apd(image)
y_apd(i_category)
self.N = len(x)
return x, np.array(y, dtype=np.int32)
def get_data_for_predict(self, ifpath, image_name, reshape_size=(3, 32, 32)):
image = np.array(Image.open(ifpath + "/" + image_name), dtype=np.float32)
image = cv2.resize(image, (reshape_size[1], reshape_size[2]))
# print image.shape
image = image.reshape(reshape_size)
return [image]
def forward(self, x_data, y_data, train=True):
x, t = Variable(np.array(x_data)), Variable(y_data)
h1 = F.max_pooling_2d(F.relu(self.model.bn1(self.model.conv1(x))), 2)
h2 = F.max_pooling_2d(F.relu(self.model.bn2(self.model.conv2(h1))), 2)
h3 = F.max_pooling_2d(F.relu(self.model.conv3(h2)), 2)
h4 = F.dropout(F.relu(self.model.fl4(h3)), train=train)
y = self.model.fl5(h4)
if train:
return F.softmax_cross_entropy(y, t), F.accuracy(y, t)
else:
res = [d for data in F.softmax(y).data for d in data]
# print res
return np.array(res).argmax() if len([r for r in res if r > 0.5]) > 0 else "unknown"
def get_num_of_images(self, path, image_name):
cmd = "ls images|grep %s|wc -l" % (image_name)
return int(subprocess.check_output(cmd, shell=True))
def dump_model(self):
self.model.to_cpu()
with open(self.model_name, "wb") as f:
pickle.dump(self.model, f, -1)
def load_model(self):
with open(self.model_name, "rb") as f:
self.model = pickle.load(f)
def fit(self, x_train, y_train, epoch=20, batchsize=100):
for epoch in xrange(1, epoch + 1):
print "epoch", epoch
# training
sum_accuracy = 0
sum_loss = 0
for i in xrange(0, self.N, batchsize):
self.optimizer.zero_grads()
loss, acc = self.forward(x_train[i : i + batchsize], y_train[i : i + batchsize])
loss.backward()
self.optimizer.update()
print "train mean loss=%s, accuracy =%s" % (str(loss.data), str(acc.data))
self.dump_model()
def predict(self, x):
y = self.forward(x, np.zeros(1, dtype=np.int32), train=False)
sys.stdout.write(str(self.file_names[y]) if y != "unknonw" else "unknonw")
