Exemplo n.º 1
0
    else:
        if not args['output_folder'] in os.listdir('../'):
            print os.listdir('../')
            os.mkdir('../' + args['output_folder'], 0755)
        output_path = "../" + args['output_folder']

    population = int(args['population'])
    input_path = args['file']

    img = Image.open(input_path)
    if args["width"] != None:
        img = setSize(int(args["width"]), img)
    size = img.size

    artist = Artist(None, 50, size)
    parent_img = artist.draw(size, (0, 0, 0, 255))
    artist.getFittness(parent_img, img)

    while (1):
        child_artist = Artist(artist.DNA, 50, size)
        child_artist.DNA = child_artist.mutate()
        child_img = child_artist.draw(size, (0, 0, 0, 255))
        child_artist.getFittness(child_img, img)
        artists.append(child_artist)

        for child_artist in artists:
            if child_artist.fittness < artist.fittness:
                artist = child_artist

        if generation % 50 == 0:
            print "[+] Generation: {}".format(generation)