-
Notifications
You must be signed in to change notification settings - Fork 0
/
beta.py
43 lines (30 loc) · 1.29 KB
/
beta.py
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
import cv2
import numpy as np
import moviepy.editor as mpy
background = cv2.imread('./Malibu.jpg')
output = []
i = 0
ratio = 360/background.shape[0]
bg = cv2.resize(background, (int(background.shape[1] * ratio), 360))
bg = cv2.flip(bg, 1)
# bg = cv2.resize(background, (0,0), fx=0.4, fy=0.5)
for i in range(180):
img_Path = f'./cat/cat_{i}.png'
img = cv2.imread(img_Path)
fliped_image = cv2.flip(img, 1)
foreground = np.logical_or(img[:,:,1] < 180, img[:,:,0]>150)
nonzero_x,nonzero_y = np.nonzero(foreground)
nonzero_cat_values = img[nonzero_x, nonzero_y,:]
foreground_1 = np.logical_or(fliped_image[:,:,1] < 180, fliped_image[:,:,0]>150)
nonzero_x1,nonzero_y1 = np.nonzero(foreground_1)
nonzero_cat_values_flipped = fliped_image[nonzero_x1, nonzero_y1,:]
new_frame = bg.copy()
new_frame[nonzero_x, nonzero_y, :] = nonzero_cat_values
fliped_frame = cv2.flip(new_frame, 1)
fliped_frame[nonzero_x, nonzero_y, :] = nonzero_cat_values
# fliped_frame = cv2.cvtColor(fliped_frame, cv2.COLOR_RGB2BGR)
output.append(fliped_frame)
clip = mpy.ImageSequenceClip(output, fps=25)
audio = mpy.AudioFileClip('./selfcontrol_part.wav').set_duration(clip.duration)
clip = clip.set_audio(audioclip = audio)
clip.write_videofile('Part-II.mp4', codec='libx264')