def screenshot(self, frame, filename=None):
self.screenshots += 1
plt.imshow(as_RGB(frame))
if filename:
plt.savefig(self.save_path + '/' + filename)
else:
plt.savefig(self.save_path + '/screen{}.png'.format(self.screenshots))
plt.clf()
def agent_cleanup(self):
print 'saving memory'
background = np.argmax(self.counts,axis=1)
with file(self.log_name+'.pkl','wb') as f:
pickle.dump(background,f,-1)
imsave(self.log_name+'.png',
as_RGB(background.reshape(self.im_size)))
def agent_cleanup(self):
print 'saving memory'
background = np.argmax(self.counts, axis=1)
with file(self.log_name + '.pkl', 'wb') as f:
pickle.dump(background, f, -1)
imsave(self.log_name + '.png',
as_RGB(background.reshape(self.im_size)))
def screenshot(self, frame, filename=None):
self.screenshots += 1
plt.imshow(as_RGB(frame))
if filename:
plt.savefig(self.save_path + '/' + filename)
else:
plt.savefig(self.save_path +
'/screen{}.png'.format(self.screenshots))
plt.clf()
