Есть ли причина, по которой этот код экспоненциально замедляется со временем, и есть ли способ это исправить? - PullRequest
1 голос
/ 07 июля 2019

Со временем эта программа начинает работать все медленнее и медленнее до такой степени, что ее практически невозможно использовать.

Я попытался удалить переменные, используемые в циклах, освободить видео, которое я использую с 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)

Этот код принимает видео, а затем выводит черно-белое видео на переднем плане исходного видео.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...