combine-images

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/background.png filter=lfs diff=lfs merge=lfs -text
+examples/foreground.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,58 @@
+import gradio as gr
+from rembg import remove
+from PIL import ImageOps, ImageDraw
+
+def process_image(foreImg, flip, backImg, left, top, width, height, isLoc):
+	if foreImg is None:
+		return None
+
+	if isLoc:
+		if backImg is None:
+			return None
+		rect = ImageDraw.Draw(backImg)
+		imwidth = foreImg.width
+		imheight = foreImg.height
+		if width > 0:
+			imwidth = width
+		if height > 0:
+			imheight = height
+		rect.rectangle([(left, top), (imwidth + left, imheight + top)], outline=(255, 0, 0), width=4)
+		return backImg
+
+	image = remove(foreImg)
+	if flip:
+		image = ImageOps.mirror(image)
+
+	imwidth = image.width
+	imheight = image.height
+	if width > 0:
+		imwidth = width
+	if height > 0:
+		imheight = height
+	image = image.resize((imwidth, imheight))
+
+	if backImg is None:
+		return image
+
+	backImg.paste(image, (left, top), image)
+	return backImg
+
+gr.Interface(
+	fn=process_image,
+	inputs=[
+		gr.Image(label='foreground', type='pil'),
+		gr.Checkbox(label='flip left and right'),
+		gr.Image(label='background', type='pil'),
+		gr.Slider(maximum=4000, step=1, label='left'),
+		gr.Slider(maximum=4000, step=1, label='top'),
+		gr.Slider(maximum=4000, step=1, label='width'),
+		gr.Slider(maximum=4000, step=1, label='height'),
+		gr.Checkbox(label='check location only'),
+	],
+	title='combine images',
+	description='combine images',
+	outputs='image',
+	allow_flagging='never',
+	examples=[['examples/foreground.png', False, 'examples/background.png', 720, 540, 256, 256, False]],
+	#cache_examples=False
+).launch()

requirements.txt

@@ -0,0 +1 @@
+rembg