Spaces:
Running
Running
| 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) | |