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