[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-26 12:07:27 +03:00
parent 44427e91ca
commit b6697e71b1

@ -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