fxreader.online/freelance-project-34-marketing-blog
2
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

[~] Refactor

Browse Source
This commit is contained in:
Siarhei Siniak 2021-08-02 13:44:11 +03:00
parent a8fcb28997
commit 661add54b0
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
python/tasks/mlb_player.py
View File

@ -1643,13 +1643,22 @@ def kernel_27():
p.wait() p.wait()
assert p.returncode == 0 assert p.returncode == 0
def kernel_28(video_path=None): def kernel_28(
video_path=None,
framrate=None,
max_seconds=None,
):
import cv2 import cv2
import tqdm import tqdm
import os import os
import subprocess import subprocess
import pprint import pprint
if framerate is None:
framerate = 4
if max_seconds is None:
max_seconds = 999999
if video_path is None: if video_path is None:
video_path = '/kaggle/working/ATL AT TOR - April 19, 2015-T0MUK91ZWys.mp4' video_path = '/kaggle/working/ATL AT TOR - April 19, 2015-T0MUK91ZWys.mp4'
t5 = video_path t5 = video_path
@ -1663,10 +1672,10 @@ def kernel_28(video_path=None):
cap = cv2.VideoCapture(t5) cap = cv2.VideoCapture(t5)
fps = cap.get(cv2.CAP_PROP_FPS) # OpenCV2 version 2 used "CV_CAP_PROP_FPS" fps = cap.get(cv2.CAP_PROP_FPS) # OpenCV2 version 2 used "CV_CAP_PROP_FPS"
frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT)) frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
duration = frame_count/fps duration = min(frame_count/fps, max_seconds)
#R = int(round(30 / 5)) #R = int(round(30 / 5))
FRAMERATE = 4 FRAMERATE = framerate
SLICE_LENGTH = 5 * 6 SLICE_LENGTH = 5 * 6
for i in tqdm.tqdm(range(int(duration / SLICE_LENGTH + 1e-8))): for i in tqdm.tqdm(range(int(duration / SLICE_LENGTH + 1e-8))):
t2 = os.path.join(t3, 'slice-%d' % i) t2 = os.path.join(t3, 'slice-%d' % i)