gui.py 文件源码

def update_canvas(self, first_time = False):

        units  = [(28,28)]  +[(10,n/10) for n in self.hidden_sizes]+[(1,10)]
        pixels = [(140,140)]+ [(n/2,50) for n in self.hidden_sizes]+[(250,25)]

        arrays = [256*layer.eval().reshape(dimensions) for layer,dimensions in zip(self.net.layers,units)]
        images = [Image.fromarray(array).resize(dimensions) for array,dimensions in zip(arrays,pixels)]
        self.imgTks = [ImageTk.PhotoImage(image) for image in images]

        [energy, cost, _] = self.net.measure()

        if first_time:
            self.img_canvas    = [self.canvas.create_image(400, (self.n_layers-k)*100, image=imgTk) for k,imgTk in enumerate(self.imgTks)]
            self.energy_canvas = self.canvas.create_text( 20, 100, anchor=W, font="Purisa", text="Energy = %.1f" % (energy))
            self.cost_canvas   = self.canvas.create_text( 20, 200, anchor=W, font="Purisa", text="Cost = %.4f"   % (cost))
        else:
            for img_canvas, imgTk in zip(self.img_canvas,self.imgTks):
                self.canvas.itemconfig(img_canvas, image=imgTk)
            self.canvas.itemconfig(self.energy_canvas, text="Energy = %.1f" % (energy))
            self.canvas.itemconfig(self.cost_canvas,   text="Cost = %.4f"   % (cost))