Со временем эта программа начинает работать все медленнее и медленнее до такой степени, что ее практически невозможно использовать.
Я попытался удалить переменные, используемые в циклах, освободить видео, которое я использую с cv2, используя cap.release (), и добавить в списки вместо использования функции np.append ().
def video_generator(name_of_video, video, cap, number_of_frames):
check, frame = cap.read()
A = []
a = []
if cap.isOpened():
width = int(cap.get(3))
height = int(cap.get(4))
dims = (height, width)
while cap.isOpened():
check, frame = cap.read()
if check:
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY).flatten()
A.append(gray)
a.append('row')
else:
break
Data = np.array(A)
B = np.resize(Data, (number_of_frames, width * height))
U, S, V = decomposition.randomized_svd(B, 2)
background = U @ np.diag(S) @ V
foreground = ([])
def generate_frames(B, background):
for i in range(number_of_frames):
foreground = np.reshape(B[i] - background[i], dims)
frame = plt.imshow(foreground, cmap = 'gray')
plt.savefig('D:/Python_Programs/BR_SVD/Frames/' + name_of_video + '/file_' + str(i + x * number_of_frames).zfill(2) + '.png')
del frame
del foreground
gc.collect()
generate_frames(B, background)
del B
del background
gc.collect()
cap.release()
if number_of_frames_initial <= 200:
cap = cv2.VideoCapture(video)
video_generator(name_of_video, video_initial, cap_initial, number_of_frames_initial)
else:
number_of_videos = int(number_of_frames_initial/200 + 1)
time_of_videos = duration/number_of_videos
for x in range (number_of_videos):
subprocess.call ([
'ffmpeg', '-i', video_initial, '-ss', str(time_of_videos * x), '-t', str(time_of_videos), 'D:/Python_Programs/BR_SVD/Temp_Videos/video_%02d.mp4' % x])
video = 'D:/Python_Programs/BR_SVD/Temp_Videos/video_%02d.mp4' % x
cap = cv2.VideoCapture(video)
number_of_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
video_generator(name_of_video, video, cap, number_of_frames)
Этот код принимает видео, а затем выводит черно-белое видео на переднем плане исходного видео.