wajjuu / pose-with-action Goto Github PK

$ pip3 install torch==1.1.0 torchvision==0.3.0

$ export PATH=/usr/local/cuda/bin/:$PATH

$ export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH

$ pip install cython

$ sudo apt-get install libyaml-dev

$ python setup.py build develop --user

$ python -m pip install Pillow==6.2.1

$ pip install -U PyYAML

$ python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir examples/demo/ --save_img

$ python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --video examples/demo/3.mp4 --outdir examples/res1 --save_video --gpus 0