[~] Refactor
This commit is contained in:
parent
888814aa64
commit
fdba7d9b29
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user