Kospo X Dacon Car Crash Analysis AI Contest(Recruiting)

def get_frames(path):
    cap = cv2.VideoCapture(path)
    frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    imgs = []        
    for fidx in range(frames):
        _, img = cap.read()            
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        imgs.append(img)
    
    return np.array(imgs)

path = train.loc[train["weather"]==3, "video_path"].values

@interact(frame=(0, 50-1), path=path)
def show_frame(frame=0, path=None):
    frames = np.zeros((1, 20, 20, 3))
    if path:
        frames = get_frames(path)
    plt.imshow(frames[frame,:,:,:])