neuro-lab6/aug-ol.py

36 lines
856 B
Python
Raw Permalink Normal View History

2025-11-29 13:57:50 +02:00
from PIL import Image
from os import listdir as ls
from random import randint as r
src = "../stock-images-part-2"
dst = "../stock-images-part-2-ovly"
olp = "../cnh-bw.png"
a = []
for k, i in enumerate(ls(src)):
print(k)
a.append(Image.open(f"{src}/{i}").resize((512,512)).convert("RGB"))
img_ol = Image.open(olp)#.resize((200,200))
for k, img in enumerate(a):
print(k)
n = img_ol.resize((r(150, 250), r(150, 250)))
match r(0, 3):
case 0:
img.paste(n, (r(0, 261), r(0, 261)))
case 1:
n = n.rotate(90)
img.paste(n, (r(0, 261), r(0, 261)))
case 2:
n = n.rotate(180)
img.paste(n, (r(0, 261), r(0, 261)))
case 3:
n = n.rotate(270)
img.paste(n, (r(0, 261), r(0, 261)))
img.resize((300,300)).save(f"{dst}/{k}.jpg")