File size: 393 Bytes
c802d72
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
#unused file
import cv2
import numpy as np

def load_video(path, size=(224,224)):
    cap = cv2.VideoCapture(path)
    frames = []

    while True:
        ret, frame = cap.read()
        if not ret:
            break

        frame = cv2.resize(frame, size)
        frame = (frame.astype(np.float32) / 255.0) * 2 - 1
        frames.append(frame)

    cap.release()
    return np.array(frames)