grey_sprite.draw() sobel_sprite.draw() hog_sprite.draw() hidden_sprite.draw() for st in str_list: st.draw() arrow.draw() k = mykeys.read() if k >-1: if k == 27: mykeys.close() DISPLAY.destroy() break elif k == ord(' '): im = np.array(Image.open(random.choice(flist))) outlayer = mlp.feed_forward(ip.get_hog(im)) grey_tex.update_ndarray(false_colour(ip.grey_scale)) sobel_tex.update_ndarray(false_colour(ip.edges)) hog_tex.update_ndarray(false_colour(ip.hog.reshape(16, 128))) hidden_tex.update_ndarray(false_colour(mlp.ah.reshape(8, 8))) arrow.positionY(1.0 - 0.5 * outlayer.argmax()) str_colours = false_colour(outlayer.reshape(5,1)) / 255.0 for i, st in enumerate(str_list): st.set_material(str_colours[i, 0]) elif k == ord('p'): pi3d.screenshot("/home/patrick/Downloads/scr_caps_pi3d/scr_caps/fr{:03d}.jpg".format(fr)) fr += 1
grey_sprite.draw() sobel_sprite.draw() hog_sprite.draw() hidden_sprite.draw() for st in str_list: st.draw() arrow.draw() k = mykeys.read() if k > -1: if k == 27: mykeys.close() DISPLAY.destroy() break elif k == ord(' '): im = np.array(Image.open(random.choice(flist))) outlayer = mlp.feed_forward(ip.get_hog(im)) grey_tex.update_ndarray(false_colour(ip.grey_scale)) sobel_tex.update_ndarray(false_colour(ip.edges)) hog_tex.update_ndarray(false_colour(ip.hog.reshape(16, 128))) hidden_tex.update_ndarray(false_colour(mlp.ah.reshape(8, 8))) arrow.positionY(1.0 - 0.5 * outlayer.argmax()) str_colours = false_colour(outlayer.reshape(5, 1)) / 255.0 for i, st in enumerate(str_list): st.set_material(str_colours[i, 0]) elif k == ord('p'): pi3d.screenshot( "/home/patrick/Downloads/scr_caps_pi3d/scr_caps/fr{:03d}.jpg". format(fr)) fr += 1