[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-08-02 14:46:21 +03:00
parent 8c78463d1b
commit 46ba4518ac

@ -1826,7 +1826,14 @@ def kernel_30(
max_frames=None, max_frames=None,
max_ids=None, max_ids=None,
video_path=None, video_path=None,
low_portion=None,
low_mean_conf=None,
): ):
if low_portion is None:
low_portion = 0.1
if low_mean_conf is None:
low_mean_conf = 0.6
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'
@ -1852,8 +1859,8 @@ def kernel_30(
elif ids == 'v1': elif ids == 'v1':
t8 = numpy.where( t8 = numpy.where(
numpy.logical_and( numpy.logical_and(
o_29['t5'].portion > 0.1, o_29['t5'].portion > low_portion,
o_29['t5'].mean_conf > 0.6 o_29['t5'].mean_conf > low_mean_conf,
) )
)[0] )[0]
ids = numpy.random.choice( ids = numpy.random.choice(
@ -1965,6 +1972,7 @@ def kernel_30(
]) ])
else: else:
assert isinstance(ids, numpy.ndarray) assert isinstance(ids, numpy.ndarray)
ids = ids[:max_ids] ids = ids[:max_ids]
pprint.pprint(['ids', ids]) pprint.pprint(['ids', ids])
@ -2349,6 +2357,8 @@ def kernel_38(
ids=None, ids=None,
prev_frames=None, prev_frames=None,
delay=None, delay=None,
low_mean_conf=None,
low_portion=None,
): ):
if ids is None: if ids is None:
ids = 'v7' ids = 'v7'
@ -2395,6 +2405,8 @@ def kernel_38(
max_frames=9999, max_frames=9999,
max_ids=999, max_ids=999,
video_path=t1, video_path=t1,
low_mean_conf=low_mean_conf,
low_portion=low_portion,
) )
return dict( return dict(