def write_log(self, params, get_output_layer, epoch, iter): # write layer1 output op_out = get_output_layer(0) l0name = '{0:s}/Output-test-epoch{1:04d}-iter{2:04d}-'.format(params.op_dir, epoch, iter) + '{0:d}.png' #tools.write_image_multiple(op_out, params.imshape[:2], params.pnt_nos, l0name) #self.representative_image = l0name tools.write_image_multiple(op_out, (1,1), params.pnt_nos, l0name)
def write_log(self, params, get_output_layer, epoch, iter): # write layer1 output op_out = numpy.asarray([get_output_layer(0)]).T l0name = '{0:s}/SofMax-test-epoch{1:04d}-iter{2:04d}-'.format(params.op_dir, epoch, iter) + '{0:d}.png' tools.write_image_multiple(op_out, (1,1), params.pnt_nos, l0name)