From cc9e76499a5fdb2e966f7b8c1603005bb7fcf603 Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Thu, 29 Jul 2021 07:59:34 +0300
Subject: [PATCH] [~] Refactor
---
python/tasks/mlb_player.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py
index 08e238d..974be25 100644
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@@ -1751,14 +1751,16 @@ def kernel_28():
t8 = numpy.array([t1[0] * fps, t1[1] * fps]).astype(numpy.int32)
cap.set(cv2.CAP_PROP_POS_FRAMES, t8[0])
+ t12 = numpy.arange(t8[0], t8[1], SLICE_LENGTH * FRAMERATE)
for k in range(t8[1] - t8[0]):
ret, frame = cap.read()
t11 = cap.get(cv2.CAP_PROP_POS_FRAMES)
- t10 = os.path.join(
- t2,
- 'frame-%d.jpg' % t11,
- )
- cv2.imwrite(t10, frame)
+ if numpy.isin(t11, t12):
+ t10 = os.path.join(
+ t2,
+ 'frame-%d.jpg' % t11,
+ )
+ cv2.imwrite(t10, frame)
t6 = '''
cd /kaggle/working/AlphaPose && \