From ad79a243d3699e1ccf88170258c2df10acac126e Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Mon, 26 Jul 2021 13:46:32 +0300
Subject: [PATCH] [~] Refactor

---
 .gitmodules                |  3 +++
 d1/mlb_player_v5.py        |  2 +-
 deps/alpha-pose            |  1 +
 python/tasks/mlb_player.py | 24 ++++++++++++++++++++++++
 4 files changed, 29 insertions(+), 1 deletion(-)
 create mode 100644 .gitmodules
 create mode 160000 deps/alpha-pose

diff --git a/.gitmodules b/.gitmodules
new file mode 100644
index 0000000..44647ff
--- /dev/null
+++ b/.gitmodules
@@ -0,0 +1,3 @@
+[submodule "deps/alpha-pose"]
+	path = deps/alpha-pose
+	url = https://github.com/WildflowerSchools/AlphaPose
diff --git a/d1/mlb_player_v5.py b/d1/mlb_player_v5.py
index bae8f27..7892741 100644
--- a/d1/mlb_player_v5.py
+++ b/d1/mlb_player_v5.py
@@ -50,5 +50,5 @@ for o1, o2 in [
 
 assert os.system(r'''
 mkdir -p /kaggle/working/test-input && mkdir -p /kaggle/working/test-output && cp /kaggle/working/AlphaPose/examples/demo/*.jpg /kaggle/working/test-input
-cd /kaggle/working/AlphaPose && python3 scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir /kaggle/working/test-input --save_img
+cd /kaggle/working/AlphaPose && python3 scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir /kaggle/working/test-input --outdir /kaggle/working/test-output --save_img
 ''') == 0
diff --git a/deps/alpha-pose b/deps/alpha-pose
new file mode 160000
index 0000000..e2e4394
--- /dev/null
+++ b/deps/alpha-pose
@@ -0,0 +1 @@
+Subproject commit e2e4394f996efd73bb8c8eb53ccfb305136ae0f6
diff --git a/python/tasks/mlb_player.py b/python/tasks/mlb_player.py
index 5f6216d..7fc910f 100644
--- a/python/tasks/mlb_player.py
+++ b/python/tasks/mlb_player.py
@@ -1154,3 +1154,27 @@ def kernel_15(
                 t23,
             ]
         )
+
+def kernel_16():
+	import argparse
+	import os
+	import platform
+	import sys
+	import time
+
+	import numpy as np
+	import torch
+	from tqdm import tqdm
+	import natsort
+
+	from detector.apis import get_detector
+	from trackers.tracker_api import Tracker
+	from trackers.tracker_cfg import cfg as tcfg
+	from trackers import track
+	from alphapose.models import builder
+	from alphapose.utils.config import update_config
+	from alphapose.utils.detector import DetectionLoader
+	from alphapose.utils.transforms import flip, flip_heatmap
+	from alphapose.utils.vis import getTime
+	from alphapose.utils.webcam_detector import WebCamDetectionLoader
+	from alphapose.utils.writer import DataWriter