Скрипт для генерации уникальных изображений?

Question

[Саша Билозир]

Здравствуйте. Подскажите существуют ли инструменты, с помощью которых можно создать подобный набор картинок? Один рисунок с разными уникальными комбинациями цветом.

Answer 1

[MinTnt]

Думаю тут и PIL сгодится. Сначала задав код для рисования по пиксельно нужного изображения. А дальше создать список с разными цветами, с которого потом через random выбирать нужный цвет.

Т.е. создать по функции для рисования определёных одноцветных мест рисунка. А далее выбирая через random случайный код света, передавать его функции, для отрисовки.

Update: я пока лишь так, быстро просмотрел PIL, написал пусть и не до конца идеальный, но для начала сойдёт - мини-код. Он берёт из папки уже существующее входное изображение, а на выходе изменённое со случайными цветами.

from PIL import Image, ImageDraw 
from random import randint
import time

image = Image.open('enter-image.png')
pix = image.load()
width = image.size[0]
height = image.size[1]
draw = ImageDraw.Draw(image)

pixel_id = []
print(f'Через 3 секунды начнётся анализ картинки. Размеры {width}x{height}')
time.sleep(3)
start_time = time.time()

for x in range(width):
	for y in range(height):
		print(x, y)
		if not pix[x,y] in [i[0] for i in pixel_id]:
			pixel_id.append([pix[x,y], []])
		pixel_id[[i[0] for i in pixel_id].index(pix[x,y])][1].append((x,y))

for a in pixel_id:
	color = (randint(0, 255), randint(0, 255) , randint(0, 255))
	for draw_zone in a[1]:
		draw.point(draw_zone, color)

print('Проработка изменённой картинки завершена. Начинается сохранение.')
image.save("output-image.png")
print(f'Сохранение завершено за {time.time()-start_time}.')

Если же нужно чтоб какие-то цвета не брало в счёт, можно приделать такую проверку.

if pix[x,y] not in [(0, 0, 0, 255), (255,255,255, 255)]:
			if not pix[x,y] in [i[0] for i in pixel_id]:
				pixel_id.append([pix[x,y], []])
			pixel_id[[i[0] for i in pixel_id].index(pix[x,y])][1].append((x,y))

Comments

[Саша Билозир]

Спасибо за ответ.
К сожалению, не знаю python и с нуля не смогу разобраться за короткое время.
Надеялся найти хоть какое-то готовое решение и попытаться поковырять для своей задачи.

[MinTnt]

Саша Билозир, сам я знаю лишь Python, а PIL лишь краем глаза видел. Но вот первая выбившая статья очень даже помогает, https://habr.com/ru/post/451074/

Сначала начиная с создания пустой картинки:

from PIL import Image, ImageDraw 
#n, m - ширина, высота(или наоборот)
image = Image.new('RGB', (n, m))

Потом из статьи выше берём

draw = ImageDraw.Draw(image)  # Создаем инструмент для рисования

Потом есть там ниже строчка:
draw.point((x, y), (255 - sr, 255 - sr, 255 - sr))
Так что можно создать массив с кортежами кординат по типу
zone_image1 = [(1, 3), (2, 5), (5, 10), ]
и потом создать функцию для отрисовки по типу

def draw_zone(zone, color):
    for i in zone:
        draw.point((i[0], i[1), color)

#Которую потом вызывать по типу
from random import randint
zone_image1 = [(1, 3), (2, 5), (5, 10)]
draw_zone(zone_image1, (randint(1, 255), randint(1, 255), randint(1, 255)))

И в конце сохраняем всё
image.save("image.jpg", "JPEG")

[MinTnt]

Саша Билозир, Хм.. я тут пока ответ писал, можно сказать чутка изучил PIL, даж интересно стало самому такое написать.

[MinTnt]

А это чуть более доработанный варянт кода:

from PIL import Image, ImageDraw 
from random import randint
import time

image = Image.open('enter-image.png')
pix = image.load()
width = image.size[0]
height = image.size[1]
draw = ImageDraw.Draw(image)

pixel_id = []
print(f'Через 3 секунды начнётся анализ картинки. Размеры {width}x{height}')
time.sleep(3)
start_time = time.time()

for x in range(width):
	for y in range(height):
		print(x, y)
		if pix[x,y] not in [(0, 0, 0, 255), (255,255,255, 255)]:
			if not pix[x,y] in [i[0] for i in pixel_id]:
				pixel_id.append([pix[x,y], []])
			pixel_id[[i[0] for i in pixel_id].index(pix[x,y])][1].append((x,y))

print('Проработка изменённой картинки завершена. Начинается сохранение.')
for i in range(int(input('Сколько вам нужно случайно сгенерированных картинок?: '))):
	for a in pixel_id:
		color = (randint(0, 255), randint(0, 255) , randint(0, 255))
		for draw_zone in a[1]:
			draw.point(draw_zone, color)

	image.save(f"output-image{i}.png")
	print(f'Картинка {i} сохранена.')
print(f'Общее время обработки фотографий {time.time()-start_time}.')

[Саша Билозир]

MinTnt, вау, большое спасибо! Буду пробовать.

[o5a]

MinTnt, возможно стоит еще

вместо исключения чисто черного и белого
if pix[x,y] not in [(0, 0, 0, 255), (255,255,255, 255)]:
исключать все оттенки серого
r,g,b,*_ = pix[x,y]
    if not (r == g == b):

Так будет выглядеть менее чужеродно, т.к. даже на границах с черной линией есть градиент. Плюс не будет захватывать серый фон (как в данном примере картинки).

[MinTnt]

o5a, а это хорошая идея. Я просто про PIL узнал лишь с первой найденной статьи.

Пока вот думаю, считается ли правильным ещё использовать потоки, а то при первом цикле с по-пиксельным прохождении создаётся некий шаблон одинаковых цветов, по которому потом и идёт отрисовка.