[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-20 08:31:20 +03:00
parent d10c5664f5
commit 6e786e3de0

@ -756,7 +756,11 @@ def kernel_7(
img_ori,
name=None,
scale_param=None,
display=None,
):
if display is None:
display = True
if scale_param is None:
scale_param = [0.5, 1.0, 1.5, 2.0]
@ -784,6 +788,7 @@ def kernel_7(
subsets, img_points = draw_key_point(subsets, peaks, img_ori)
# After predicting Heatmaps and PAF's, proceeed to link joints correctly
if display:
img_canvas = link_key_point(img_points, candidates, subsets)
@ -821,7 +826,7 @@ def kernel_8(
def kernel_9_benchmark(
o_7,
):
t1 = o_7['cv2'].imread('../input/indonesian-traditional-dance/tgagrakanyar/tga_0000.jpg'
t1 = o_7['cv2'].imread('../input/indonesian-traditional-dance/tgagrakanyar/tga_0000.jpg')
t5 = 10
t2 = datetime.datetime.now()
for k in range(t5):