From 8b91f62fa783e4ce576e32e2ea1d8a8b3bc6a6ac Mon Sep 17 00:00:00 2001 From: Siarhei Siniak Date: Tue, 20 Jul 2021 08:33:53 +0300 Subject: [PATCH] [~] Refactor --- python/tasks/mlb_player.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py index 8ba8297..6465457 100644 --- a/python/tasks/mlb_player.py +++ b/python/tasks/mlb_player.py @@ -764,14 +764,15 @@ def kernel_7( if scale_param is None: scale_param = [0.5, 1.0, 1.5, 2.0] - if name is None: - name = tempfile.mktemp( - dir='/kaggle/working', - suffix='.png', + if display: + if name is None: + name = tempfile.mktemp( + dir='/kaggle/working', + suffix='.png', + ) + pprint.pprint( + ['estimate_pose', dict(name=name)], ) - pprint.pprint( - ['estimate_pose', dict(name=name)], - ) # People might be at different scales in the image, perform inference at multiple scales to boost results