Пример #1
0
 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)
Пример #2
0
 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)