def vanilla(): from PIL import Image target_path = "..\\img\\target_diamond.png" target_path = "..\\img\\target_spade.png" # target_path = "..\\img\\target_mickey.png" # target_path = "..\\img\\target_heart.png" # target_path = "..\\img\\target_forward.png" # target_path = "..\\img\\target_recycle.png" # target_img = Image.open(target_path).convert('L') penalties = [Optimizer.penalty_name] # energies = ["first moments (normalized)", "XOR comparison", "secondary moments (normalized)"] # energies = ["XOR comparison"] energies = ["uncovered pixels", "distance field"] # energies = ["first moments (normalized)"] # energies = ["secondary moments (normalized)"] # energy_pairs = [energies, [1e-4,1,1e-4]] weights = [1,1] energy_pairs = [energies, weights] rendis = Renderer_dispatcher() rendis.target_image = Image.open(target_path) rendis.start() init_X_Y(640, 480) plotter = plotting.Plotter(*get_fname("..\\res")) optimizer = Optimizer(rendis) plotting.attach_plotter(optimizer, plotter) optimizer.set_target(target_path) optimizer.set_method("CMA") # optimizer.set_energy(["first moments (normalized)"], [1]) optimizer.set_energy(*energy_pairs) # optimizer.set_energy(["XOR comparison"], [1]) gui = Minimalistic_GUI(optimizer) gui.run()
def all_single_energy_combos(): renderer = Renderer() renderer.start() renderer.wait_till_init() x_0 = renderer.get_param() for method_name, energy_name in product(Optimizer.method_dic, Optimizer.energy_dic): print method_name, energy_name renderer.set_param(x_0) optimizer = Optimizer(None) plotting.attach_plotter(optimizer, plotting.Plotter(*get_fname("..\\res"))) optimizer.set_target("C:\\Users\\cxz\\Pictures\\target.png") optimizer.set_method(method_name) optimizer.set_energy([energy_name], [1]) optimizer.run()