개발 공부/딥러닝

[Pose detection] - gluon_CV install & 활용

sunjungAn 2021. 4. 8. 14:50

reference : cv.gluon.ai/install.html

 

Installation — gluoncv 0.11.0 documentation

 

cv.gluon.ai

reference: github.com/dmlc/gluon-cv

 

dmlc/gluon-cv

Gluon CV Toolkit. Contribute to dmlc/gluon-cv development by creating an account on GitHub.

github.com

아래와 같이 cmd로 입력해 패키지를 받아준다. 


 

pip install --upgrade mxnet

# for pytorch pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

pip install gluoncv


from gluoncv import model_zoo, data, utils

print(model_zoo.get_model_list())

위와 같이 결과가 나온다면 성공적으로 설치를 한것이다. 

 

 

 


reference: 

cv.gluon.ai/build/examples_pose/demo_simple_pose.html#sphx-glr-download-build-examples-pose-demo-simple-pose-py

 

1. Predict with pre-trained Simple Pose Estimation models — gluoncv 0.11.0 documentation

 

cv.gluon.ai

1. pose detection에 사용될 모델 다운

from gluoncv import model_zoo, data, utils

detector = model_zoo.get_model('yolo3_mobilenet1.0_coco', pretrained=True)
pose_net = model_zoo.get_model('simple_pose_resnet18_v1b', pretrained=True)

detector.reset_class(["person"], reuse_weights=['person'])

 

2. 활용 모습

 

3. webcam detection source code : 

github.com/sunjungAn/Opencv_practice/blob/master/webcam_pose_detector(ver.1.2).py

 

sunjungAn/Opencv_practice

2021.03.01~ . Contribute to sunjungAn/Opencv_practice development by creating an account on GitHub.

github.com