Alphapose windows demo시작하기

Alphapose 는 cpu only가 안되는 것 같다..

 

1. windows부분 따라하기

https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/INSTALL.md#Windows

MVIG-SJTU/AlphaPose

2. python setup.py build develop 실행

이 때,  setup.py  파일에서

if platform.system() != 'Windows':

이 부분을

if platform.system() == 'Windows':

이렇게 고쳐주어야

deconv_conv_cuda import error가 발생하지 않음 -> cuda 필수

 

3. 무사히 빌드되었다면

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 --outdir examples/res

 

데모파일 실행

여기서 사용한 모델은 그냥 fastpose모델임

 

 detector.py에서 scipy.misc는 imread없다고? 나오는데

파일 들어가서

orig_img_k = scipy.misc.imread(im_name_k, mode='RGB')

이 부분을

orig_img_k = imageio.imread(im_name_k, pilmode='RGB')

이렇게 수정한 후 다시 위 데모 명령어 실행

 

3. 끝!

 

만약 oom이 뜨면 배치사이즈 조절해주면 된다. 조절방법은

https://github.com/MVIG-SJTU/AlphaPose/blob/master/docs/GETTING_STARTED.md

MVIG-SJTU/AlphaPose

참고