[~] Refactor

This commit is contained in:
Siarhei Siniak 2021-07-26 11:48:01 +03:00
parent 888814aa64
commit fdba7d9b29

@ -1,22 +1,26 @@
! pip install pyyaml==5.2 assert os.system(r'''
! pip install scipy==1.1.0 rm -fr /kaggle/working/AlphaPose
#! pip install torch==1.2.0 torchvision==0.4.0 pip install pyyaml==5.2
#! pip install pillow==6.2.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 import torch
print(torch.__version__) print(torch.__version__)
import yaml, scipy import yaml, scipy
print(yaml.__version__) print(yaml.__version__)
print(scipy.__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 gdown
import os import os
for o1, o2 in [ for o1, o2 in [
@ -39,14 +43,3 @@ for o1, o2 in [
o2, o2,
quiet=False 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