remove GIF
This commit is contained in:
Qing
parent
61d56288a5
commit
0e5e16ba20
@ -162,30 +162,12 @@ def parse_args():
|
||||
type=str,
|
||||
choices=RESTOREFORMER_AVAILABLE_DEVICES,
|
||||
)
|
||||
parser.add_argument(
|
||||
"--enable-gif",
|
||||
action="store_true",
|
||||
help=GIF_HELP,
|
||||
)
|
||||
parser.add_argument(
|
||||
"--install-plugins-package",
|
||||
action="store_true",
|
||||
)
|
||||
#########
|
||||
|
||||
### useless args ###
|
||||
parser.add_argument("--sd-local-model-path", default=None, help=argparse.SUPPRESS)
|
||||
parser.add_argument("--debug", action="store_true", help=argparse.SUPPRESS)
|
||||
parser.add_argument("--hf_access_token", default="", help=argparse.SUPPRESS)
|
||||
parser.add_argument(
|
||||
"--sd-disable-nsfw", action="store_true", help=argparse.SUPPRESS
|
||||
)
|
||||
parser.add_argument("--sd-run-local", action="store_true", help=argparse.SUPPRESS)
|
||||
parser.add_argument(
|
||||
"--sd-enable-xformers", action="store_true", help=argparse.SUPPRESS
|
||||
)
|
||||
### end useless args ###
|
||||
|
||||
args = parser.parse_args()
|
||||
# collect system info to help debug
|
||||
dump_environment_info()
|
||||
|
||||
@ -3,5 +3,4 @@ from .remove_bg import RemoveBG
|
||||
from .realesrgan import RealESRGANUpscaler
|
||||
from .gfpgan_plugin import GFPGANPlugin
|
||||
from .restoreformer import RestoreFormerPlugin
|
||||
from .gif import MakeGIF
|
||||
from .anime_seg import AnimeSeg
|
||||
|
||||
@ -1,149 +0,0 @@
|
||||
import io
|
||||
import math
|
||||
|
||||
from PIL import Image, ImageDraw
|
||||
|
||||
from lama_cleaner.helper import load_img
|
||||
from lama_cleaner.plugins.base_plugin import BasePlugin
|
||||
|
||||
|
||||
def keep_ratio_resize(img, size, resample=Image.BILINEAR):
|
||||
if img.width > img.height:
|
||||
w = size
|
||||
h = int(img.height * size / img.width)
|
||||
else:
|
||||
h = size
|
||||
w = int(img.width * size / img.height)
|
||||
return img.resize((w, h), resample)
|
||||
|
||||
|
||||
def cubic_bezier(p1, p2, duration: int, frames: int):
|
||||
"""
|
||||
|
||||
Args:
|
||||
p1:
|
||||
p2:
|
||||
duration: Total duration of the curve
|
||||
frames:
|
||||
|
||||
Returns:
|
||||
|
||||
"""
|
||||
x0, y0 = (0, 0)
|
||||
x1, y1 = p1
|
||||
x2, y2 = p2
|
||||
x3, y3 = (1, 1)
|
||||
|
||||
def cal_y(t):
|
||||
return (
|
||||
math.pow(1 - t, 3) * y0
|
||||
+ 3 * math.pow(1 - t, 2) * t * y1
|
||||
+ 3 * (1 - t) * math.pow(t, 2) * y2
|
||||
+ math.pow(t, 3) * y3
|
||||
)
|
||||
|
||||
def cal_x(t):
|
||||
return (
|
||||
math.pow(1 - t, 3) * x0
|
||||
+ 3 * math.pow(1 - t, 2) * t * x1
|
||||
+ 3 * (1 - t) * math.pow(t, 2) * x2
|
||||
+ math.pow(t, 3) * x3
|
||||
)
|
||||
|
||||
res = []
|
||||
for t in range(0, 1 * frames, duration):
|
||||
t = t / frames
|
||||
res.append((cal_x(t), cal_y(t)))
|
||||
|
||||
res.append((1, 0))
|
||||
return res
|
||||
|
||||
|
||||
def make_compare_gif(
|
||||
clean_img: Image.Image,
|
||||
src_img: Image.Image,
|
||||
max_side_length: int = 600,
|
||||
splitter_width: int = 5,
|
||||
splitter_color=(255, 203, 0, int(255 * 0.73)),
|
||||
):
|
||||
if clean_img.size != src_img.size:
|
||||
clean_img = clean_img.resize(src_img.size, Image.BILINEAR)
|
||||
|
||||
duration_per_frame = 20
|
||||
num_frames = 50
|
||||
# erase-in-out
|
||||
cubic_bezier_points = cubic_bezier((0.33, 0), (0.66, 1), 1, num_frames)
|
||||
cubic_bezier_points.reverse()
|
||||
|
||||
max_side_length = min(max_side_length, max(clean_img.size))
|
||||
|
||||
src_img = keep_ratio_resize(src_img, max_side_length)
|
||||
clean_img = keep_ratio_resize(clean_img, max_side_length)
|
||||
width, height = src_img.size
|
||||
|
||||
# Generate images to make Gif from right to left
|
||||
images = []
|
||||
|
||||
for i in range(num_frames):
|
||||
new_frame = Image.new("RGB", (width, height))
|
||||
new_frame.paste(clean_img, (0, 0))
|
||||
|
||||
left = int(cubic_bezier_points[i][0] * width)
|
||||
cropped_src_img = src_img.crop((left, 0, width, height))
|
||||
new_frame.paste(cropped_src_img, (left, 0, width, height))
|
||||
if i != num_frames - 1:
|
||||
# draw a yellow splitter on the edge of the cropped image
|
||||
draw = ImageDraw.Draw(new_frame)
|
||||
draw.line(
|
||||
[(left, 0), (left, height)], width=splitter_width, fill=splitter_color
|
||||
)
|
||||
images.append(new_frame)
|
||||
|
||||
for i in range(30):
|
||||
images.append(src_img)
|
||||
|
||||
cubic_bezier_points.reverse()
|
||||
# Generate images to make Gif from left to right
|
||||
for i in range(num_frames):
|
||||
new_frame = Image.new("RGB", (width, height))
|
||||
new_frame.paste(src_img, (0, 0))
|
||||
|
||||
right = int(cubic_bezier_points[i][0] * width)
|
||||
cropped_src_img = clean_img.crop((0, 0, right, height))
|
||||
new_frame.paste(cropped_src_img, (0, 0, right, height))
|
||||
if i != num_frames - 1:
|
||||
# draw a yellow splitter on the edge of the cropped image
|
||||
draw = ImageDraw.Draw(new_frame)
|
||||
draw.line(
|
||||
[(right, 0), (right, height)], width=splitter_width, fill=splitter_color
|
||||
)
|
||||
images.append(new_frame)
|
||||
|
||||
for _ in range(30):
|
||||
images.append(clean_img)
|
||||
|
||||
img_byte_arr = io.BytesIO()
|
||||
clean_img.save(
|
||||
img_byte_arr,
|
||||
format="GIF",
|
||||
save_all=True,
|
||||
include_color_table=True,
|
||||
append_images=images,
|
||||
optimize=False,
|
||||
duration=duration_per_frame,
|
||||
loop=0,
|
||||
)
|
||||
return img_byte_arr.getvalue()
|
||||
|
||||
|
||||
class MakeGIF(BasePlugin):
|
||||
name = "MakeGIF"
|
||||
|
||||
def __call__(self, rgb_np_img, files, form):
|
||||
origin_image = rgb_np_img
|
||||
clean_image_bytes = files["clean_img"].read()
|
||||
clean_image, _ = load_img(clean_image_bytes)
|
||||
gif_bytes = make_compare_gif(
|
||||
Image.fromarray(origin_image), Image.fromarray(clean_image)
|
||||
)
|
||||
return gif_bytes
|
||||
@ -33,7 +33,6 @@ from lama_cleaner.plugins import (
|
||||
InteractiveSeg,
|
||||
RemoveBG,
|
||||
RealESRGANUpscaler,
|
||||
MakeGIF,
|
||||
GFPGANPlugin,
|
||||
RestoreFormerPlugin,
|
||||
AnimeSeg,
|
||||
@ -373,13 +372,6 @@ def run_plugin():
|
||||
logger.info(f"{name} process time: {(time.time() - start) * 1000}ms")
|
||||
torch_gc()
|
||||
|
||||
if name == MakeGIF.name:
|
||||
return send_file(
|
||||
io.BytesIO(bgr_res),
|
||||
mimetype="image/gif",
|
||||
as_attachment=True,
|
||||
download_name=form["filename"],
|
||||
)
|
||||
if name == InteractiveSeg.name:
|
||||
return make_response(
|
||||
send_file(
|
||||
@ -427,6 +419,7 @@ def get_server_config():
|
||||
"enableAutoSaving": enable_auto_saving,
|
||||
"enableControlnet": model.sd_controlnet,
|
||||
"controlnetMethod": model.sd_controlnet_method,
|
||||
"disableModelSwitch": is_disable_model_switch,
|
||||
}, 200
|
||||
|
||||
|
||||
@ -529,11 +522,6 @@ def build_plugins(args):
|
||||
upscaler=plugins.get(RealESRGANUpscaler.name, None),
|
||||
)
|
||||
|
||||
if args.enable_gif:
|
||||
logger.info(f"Initialize GIF plugin")
|
||||
plugins[MakeGIF.name] = MakeGIF()
|
||||
|
||||
|
||||
def main(args):
|
||||
global model
|
||||
global device
|
||||
|
||||
Loading…
Reference in New Issue
Block a user