File size: 2,390 Bytes
585c7ea |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import cv2, os
import numpy as np
from tqdm import tqdm
import uuid
from src.utils.videoio import save_video_with_watermark
def paste_pic(video_path, pic_path, crop_info, new_audio_path, full_video_path, extended_crop=False):
if not os.path.isfile(pic_path):
raise ValueError('pic_path must be a valid path to video/image file')
elif pic_path.split('.')[-1] in ['jpg', 'png', 'jpeg']:
# loader for first frame
full_img = cv2.imread(pic_path)
else:
# loader for videos
video_stream = cv2.VideoCapture(pic_path)
fps = video_stream.get(cv2.CAP_PROP_FPS)
full_frames = []
while 1:
still_reading, frame = video_stream.read()
if not still_reading:
video_stream.release()
break
break
full_img = frame
frame_h = full_img.shape[0]
frame_w = full_img.shape[1]
video_stream = cv2.VideoCapture(video_path)
fps = video_stream.get(cv2.CAP_PROP_FPS)
crop_frames = []
while 1:
still_reading, frame = video_stream.read()
if not still_reading:
video_stream.release()
break
crop_frames.append(frame)
if len(crop_info) != 3:
print("you didn't crop the image")
return
else:
r_w, r_h = crop_info[0]
clx, cly, crx, cry = crop_info[1]
lx, ly, rx, ry = crop_info[2]
lx, ly, rx, ry = int(lx), int(ly), int(rx), int(ry)
# oy1, oy2, ox1, ox2 = cly+ly, cly+ry, clx+lx, clx+rx
# oy1, oy2, ox1, ox2 = cly+ly, cly+ry, clx+lx, clx+rx
if extended_crop:
oy1, oy2, ox1, ox2 = cly, cry, clx, crx
else:
oy1, oy2, ox1, ox2 = cly+ly, cly+ry, clx+lx, clx+rx
tmp_path = str(uuid.uuid4())+'.mp4'
out_tmp = cv2.VideoWriter(tmp_path, cv2.VideoWriter_fourcc(*'MP4V'), fps, (frame_w, frame_h))
for crop_frame in tqdm(crop_frames, 'seamlessClone:'):
p = cv2.resize(crop_frame.astype(np.uint8), (ox2-ox1, oy2 - oy1))
mask = 255*np.ones(p.shape, p.dtype)
location = ((ox1+ox2) // 2, (oy1+oy2) // 2)
gen_img = cv2.seamlessClone(p, full_img, mask, location, cv2.NORMAL_CLONE)
out_tmp.write(gen_img)
out_tmp.release()
save_video_with_watermark(tmp_path, new_audio_path, full_video_path, watermark=False)
os.remove(tmp_path)
|