diff --git a/d1/mlb_player_v5.py b/d1/mlb_player_v5.py index ee60f78..bae8f27 100644 --- a/d1/mlb_player_v5.py +++ b/d1/mlb_player_v5.py @@ -1,20 +1,19 @@ import os -t1 = r''' -rm -fr /kaggle/working/AlphaPose -pip install pyyaml==5.2 -pip install scipy==1.1.0 -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 -''' +if os.system(r''' pip show alphapose''') != 0: + t1 = r''' + rm -fr /kaggle/working/AlphaPose + pip install pyyaml==5.2 + pip install scipy==1.1.0 + 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 + ''' -for o in t1.splitlines(): - print(o) - assert os.system(o) == 0 + for o in t1.splitlines(): + print(o) + assert os.system(o) == 0 import os #!git clone https://github.com/MVIG-SJTU/AlphaPose.git @@ -47,3 +46,9 @@ for o1, o2 in [ o2, quiet=False ) + + +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 +''') == 0