СОЗДАЕМ ИЗОБРАЖЕНИЯ ДЛЯ NFT С ПОМОЩЬЮ PYTHON (1 ЧАСТЬ)
Данная серия статей посвящена NFT, их созданию с точки зрения больших коллекций. В данной статье мы напишем простой генератор изображений на языке программирования Python. Я использую IDE VSCode и плагин notebook для набросков вроде этого.
Для работы с изображениями есть простенькая бибилиотека Pillow с которой мы и будем работать.
Первым делом создадим окружение, это не обязательный шаг, но крайне рекомендуемый и активируем его
python3 -m venv .venv source .venv/bin/activate
Установим необходимые для работы пакеты для работы
pip3 install pillow numpy
Создаим файл generate.py и запишем следующий код:
generate.pyfrom PIL import Image import os import random import math import numpy as np # Функция нормализации блока мозаики def getFixBoxSize(a, b): for i in range(0, b+1)[::-1]: if a % i == 0: return i # Размер входящего изображения in_w, in_h = (500,500) # Размер исходящего изображения out_w, out_h = (500,500) # Размер блока мозаики b_w, b_h = (100,50) # Кол-во исходных изображений use_src_images = 10 # Загружаем список изображений из директории tmp_images = os.listdir("./img") # Перемешиваем список tmp_images = sorted(tmp_images, key=lambda A: random.random()) # Оставляем нужное кол-во изображений if len(tmp_images) > use_src_images: path_images = [tmp_images[x] for x in range(0, use_src_images)] # Загружаем изображения for img in path_images: images.append(Image.open(f"./img/{img}").resize((in_w,in_h))) # Создаем новый холст mosaic = Image.new("RGB", (in_w, in_h), tuple(np.random.choice(range(256), size=3))) print(f"Source Images: " + ", ".join(path_images)) print(f"IN size image: {in_w}x{in_h}") print(f"Out size image: {out_w}x{out_h}") print(f"Block size image: {b_w}x{b_h}") b_w = getFixBoxSize(in_w, b_w) b_h = getFixBoxSize(in_h, b_h) print(f"Block FIX size image: {b_w}x{b_h}") cols, rows = (math.ceil(in_w / b_w), math.ceil(in_h/b_h)) print(f"rows: {rows}") print(f"cols: {cols}") # формируем мозаику в виде матрицы block_lines = [sorted([x2 for x2 in range(0,cols)], key=lambda A: random.random()) for x in range(0, rows)] print(" MATRIX") print("\n".join(str(x) for x in block_lines)) # Формируем в цикле наши блоки из исходных изображений for row in range(0, rows): for col in range(0, cols): rand = random.randrange(0 , len(images)) num = block_lines[row].pop(0) quad = (b_w * num, b_h * row, b_w * num + b_w, b_h * row + b_h) quadimg = images[rand].crop(quad) mosaic.paste(quadimg, (b_w * num, b_h * row, b_w * num + b_w, b_h * row + b_h), quadimg) # Изменяем размер исходящего изображения mosaic.resize((out_w, out_h)) # Выводим изображение mosaic.show()
Как видите, код получился очень простым, благодаря параметрам объявленным в самом начале, мы можешь изменять размерность блоков тем самым формируя новые стили генерации.
В следующей статье, мы перепишем данный пример и добавим генерацию нескольких изображений в коллаж, генерацию гиф анимации и что нибудь еще 🙂