[~] Refactor
This commit is contained in:
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(
|
||||||
|
Loading…
Reference in New Issue
Block a user