From fdba7d9b29a2c2b2325d4562b63e1f1a3b5b0cfd Mon Sep 17 00:00:00 2001
From: Siarhei Siniak <siarheisiniak@yahoo.com>
Date: Mon, 26 Jul 2021 11:48:01 +0300
Subject: [PATCH] [~] Refactor
---
d1/mlb_player_v5.py | 41 +++++++++++++++++------------------------
1 file changed, 17 insertions(+), 24 deletions(-)
diff --git a/d1/mlb_player_v5.py b/d1/mlb_player_v5.py
index 4d652a2..202b2bd 100644
--- a/d1/mlb_player_v5.py
+++ b/d1/mlb_player_v5.py
@@ -1,22 +1,26 @@
-! pip install pyyaml==5.2
-! pip install scipy==1.1.0
-#! pip install torch==1.2.0 torchvision==0.4.0
-#! pip install pillow==6.2.2
+assert os.system(r'''
+rm -fr /kaggle/working/AlphaPose
+pip install pyyaml==5.2
+pip install scipy==1.1.0
+pip install torch==1.2.0 torchvision==0.4.0
+pip install pillow==6.2.2
+git clone https://github.com/WildflowerSchools/AlphaPose
+python -m pip install cython gdown
+apt-get install libyaml-dev
+cd /kaggle/working/AlphaPose && python setup.py build develop
+mkdir -p /kaggle/working/test-input && mkdir -p /kaggle/working/test-output && cp 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
+''') == 0
+
+import os
+#!git clone https://github.com/MVIG-SJTU/AlphaPose.git
+
import torch
print(torch.__version__)
import yaml, scipy
print(yaml.__version__)
print(scipy.__version__)
-
-import os
-#!git clone https://github.com/MVIG-SJTU/AlphaPose.git
-!git clone https://github.com/WildflowerSchools/AlphaPose
-!python -m pip install cython gdown
-!apt-get install libyaml-dev
-
-! cd /kaggle/working/AlphaPose && python setup.py build develop
-
import gdown
import os
for o1, o2 in [
@@ -39,14 +43,3 @@ for o1, o2 in [
o2,
quiet=False
)
-
-
-import os
-os.chdir('/kaggle/working/AlphaPose')
-! ls
-! mkdir -p /kaggle/working/test-input && mkdir -p /kaggle/working/test-output && cp examples/demo/*.jpg /kaggle/working/test-input
-#! cd /kaggle/working/AlphaPose && python3 scripts/demo_inference.py --outdir /kagle/working/test-output --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 --save_img
-# result json and rendered images are saved here:
-! ls examples/res/
-! ls examples/res/vis