우주의WouldYou

http://cocodataset.org/#home

OpenPose 논문 정리

github: https://github.com/CMU-Perceptual-Computing-Lab/openpose

paper: https://arxiv.org/pdf/1611.08050.pdf

author slides: http://image-net.org/challenges/talks/2016/Multi-person%20pose%20estimation-CMU.pdf

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

1. windows부분 따라하기

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

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

참고

params = dict()
params["model_folder"] = "../../../models/"
params["process_real_time"] = True
params["video"] = "C:........" 
params["camera"] = 0
params["process_real_time"] = True
params["frames_repeat"] = True
params["body"] = 1
opWrapper = op.WrapperPython(3)
opWrapper.configure(params)
opWrapper.execute()

1. opWrapper.start()대신 execute()

2. op.WrapperPython(3)써주기 -> mode설정 필