Преобразовать изображение черепахи в python в 2D бинарную матрицу? - PullRequest
Новая
       14

Преобразовать изображение черепахи в python в 2D бинарную матрицу?

0 голосов
/ 22 декабря 2018

Я генерирую фрактальные изображения, в частности кривые Коха, используя черепаху в Python.Я намерен вычислить лакунарность изображения, для которого мне нужно это изображение в виде двумерной двоичной матрицы.

Как преобразовать вывод этой черепахи в двумерную двоичную матрицу из 0 и 1, которую я могу использовать для дальнейших вычислений?

1 Ответ

0 голосов
/ 22 декабря 2018

Сценарий ниже создает небольшое приложение tkinter, которое рисует что-то с графикой черепахи при нажатии кнопки Draw .При нажатии кнопки Сохранить графика сохраняется в файл растрового изображения.

Для преобразования графического холста черепахи в изображение RGB используется модуль PIL (библиотека изображений Python), ипреобразует это в растровое изображение размером 1 бит на пиксель.

Вот файл растрового изображения turtlescreen.png, который он создает:

turtlescreen.png image file

Вы никогда не указывали, какой формат изображения вы хотите, поэтому вам может потребоваться изменить сценарий, чтобы создать какой-то другой формат - в настоящее время он создает изображение в формате .png.Расширение имени файла изображения определяет это. 

from PIL import ImageGrab
import tkinter as tk
import tkinter.messagebox as tkMessageBox
import turtle

WIDTH, HEIGHT = 500, 400
IMG_FILENAME = 'turtlescreen.png'  # Extension determines file format.


def draw_stuff(canvas):
    screen = turtle.TurtleScreen(canvas)

    t = turtle.RawTurtle(screen.getcanvas())
    t.speed(0)  # Fastest.
    t.pencolor('black')
    t.hideturtle()

    # Draw a snowflake.
    size = 10

    def branch(size):
        for i in range(3):
            for i in range(3):
                t.forward(10.0*size/3)
                t.backward(10.0*size/3)
                t.right(45)
            t.left(90)
            t.backward(10.0*size/3)
            t.left(45)
        t.right(90)
        t.forward(10.0*size)

    # move the pen into starting position
    t.penup()
    t.forward(10*size)
    t.left(45)
    t.pendown()

    # Draw 8 branches.
    for i in range(8):
        branch(size)
        t.left(45)

    print('done')

def getter(root, widget):
    x = root.winfo_rootx() + widget.winfo_x()
    y = root.winfo_rooty() + widget.winfo_y()
    x1 = x + widget.winfo_width()
    y1 = y + widget.winfo_height()
    return ImageGrab.grab().crop((x, y, x1, y1))

def save_file(root, canvas, filename):
    """ Convert the Canvas widget into a bitmapped image. """
    # Get image of Canvas and convert it to bitmapped image.
    img = getter(root, canvas).convert('L').convert('1')
    img.save(IMG_FILENAME)  # Save image file.

    tkMessageBox.showinfo("Info", "Image saved as %r" % filename, parent=root)


# main
root = tk.Tk()
canvas = tk.Canvas(root, width=WIDTH, height=HEIGHT,
                   borderwidth=0, highlightthickness=0)
canvas.pack()

btn_frame = tk.Frame(root)
btn_frame.pack()
btn1 = tk.Button(btn_frame, text='Draw', command=lambda: draw_stuff(canvas))
btn1.pack(side=tk.LEFT)
btn2 = tk.Button(btn_frame, text='Save',
                 command=lambda: save_file(root, canvas, IMG_FILENAME))
btn2.pack(side=tk.LEFT)
btn3 = tk.Button(btn_frame, text='Quit', command=root.quit)
btn3.pack(side=tk.LEFT)

root.mainloop()
...