def step(self, u_t, h_tm1, W, W_in, W_out, b_in, b_out):
h_t1 = T.nnet.relu(T.dot(u_t, W_in) + b_in, alpha=0.2)
h_t = T.nnet.sigmoid(h_t1 + T.dot(h_tm1, W))
y_t = T.nnet.sigmoid(T.dot(h_t, W_out) + b_out)
return h_t, y_t
if __name__ == '__main__':
vr = VR('../data/person01_boxing_d4_uncomp.avi')
video_inp = T.dmatrix("video_inp")
input = []
for i in range(0, 20):
input.append(
numpy.asarray(cv2.resize(vr.next_frame(), (80, 60)).flatten()))
print numpy.min(input), numpy.max(input)
new_matrix = (input - numpy.min(input)) / (numpy.max(input) -
numpy.min(input) * 1.0)
input = new_matrix.T
print input
disc = Discriminator(memory=1000,
time_step=20,
h_size=500,
n_hid=1000,
mlp_hid=500,
lr=0.001,
video_size=80 * 60,
video_inp=video_inp)
print disc.get_auto(input).shape
print disc.train_auto(input)
self.total_params = [W,W_in,W_out]+self.gp_mlp.params+self.gen.params+[b_in,b,b_out]
print "Discrminator built"
def step(self,u_t, h_tm1, W, W_in, W_out,b_in,b_out):
h_t1 = T.nnet.relu(T.dot(u_t, W_in)+b_in,alpha=0.2)
h_t = T.nnet.sigmoid(h_t1+ T.dot(h_tm1, W))
y_t = T.nnet.sigmoid(T.dot(h_t, W_out)+b_out)
return h_t, y_t
if __name__=='__main__':
vr = VR('../data/person01_boxing_d4_uncomp.avi')
video_inp = T.dmatrix("video_inp")
input=[]
for i in range(0,20):
input.append(numpy.asarray(cv2.resize(vr.next_frame(),(80,60)).flatten()))
print numpy.min(input),numpy.max(input)
new_matrix = (input - numpy.min(input))/(numpy.max(input)-numpy.min(input)*1.0)
input = new_matrix.T
print input
disc = Discriminator(memory=1000,time_step=20,h_size=500,n_hid=1000,mlp_hid=500,lr=0.001,video_size=80*60,video_inp=video_inp)
print disc.get_auto(input).shape
print disc.train_auto(input)
print disc.train_disc(input)
print disc.train_gp()
for i in range(1,1000):
print disc.train_auto(input)
print disc.train_disc(input)
print disc.train_gp()
print disc.train_auto(input)
print disc.train_disc(input)
