CMU-Perceptual-Computing-Lab/openpose_开源项目

`Linux`

OpenPose represents the first real-time multi-person system to jointly detect human body, hand, facial, and foot keypoints (in total 135 keypoints) on single images.

Features

Functionality:
- 2D real-time multi-person keypoint detection:
  - 15 or 18 or 25-keypoint body/foot keypoint estimation. Running time invariant to number of detected people.
  - 2x21-keypoint hand keypoint estimation. Currently, running time depends on number of detected people.
  - 70-keypoint face keypoint estimation. Currently, running time depends on number of detected people.
- 3D real-time single-person keypoint detection:
  - 3-D triangulation from multiple single views.
  - Synchronization of Flir cameras handled.
  - Compatible with Flir/Point Grey cameras, but provided C++ demos to add your custom input.
- Calibration toolbox:
  - Easy estimation of distortion, intrinsic, and extrinsic camera parameters.
- Single-person tracking for further speed up or visual smoothing.
Input: Image, video, webcam, Flir/Point Grey and IP camera. Included C++ demos to add your custom input.
Output: Basic image + keypoint display/saving (PNG, JPG, AVI, ...), keypoint saving (JSON, XML, YML, ...), and/or keypoints as array class.
OS: Ubuntu (14, 16), Windows (8, 10), Mac OSX, Nvidia TX2.
Others:
- Available: command-line demo, C++ wrapper, and C++ API.
- Python API.
- Unity Plugin.
- CUDA (Nvidia GPU), OpenCL (AMD GPU), and CPU versions.

Latest Features

Jan 2018: Unity plugin released!
Jan 2018: Improved Python API released! Including body, face, hands, and all the functionality of the C++ API!
Dec 2018: Foot dataset and new paper released!
Sep 2018: Experimental single-person tracker for further speed up or visual smoothing!
Jun 2018: Combined body-foot model released! 40% faster and 5% more accurate!
Jun 2018: OpenCL/AMD graphic card version released!
Jun 2018: Calibration toolbox released!

For further details, check all released features and release notes.

Results

Body and Foot Estimation

3-D Reconstruction Module (Body, Foot, Face, and Hands)

Body, Foot, Face, and Hands Estimation

Unity Plugin

Runtime Analysis

Inference time comparison between the 3 available pose estimation libraries: OpenPose, Alpha-Pose (fast Pytorch version), and Mask R-CNN:

This analysis was performed using the same images for each algorithm and a batch size of 1. Each analysis was repeated 1000 times and then averaged. This was all performed on a system with a Nvidia 1080 Ti and CUDA 8. Megvii (Face++) and MSRA GitHub repositories were excluded because they only provide pose estimation results given a cropped person. However, they suffer the same problem than Alpha-Pose and Mask R-CNN, their runtimes grow linearly with the number of people.

Installation, Reinstallation and Uninstallation

Windows portable version: Simply download and use the latest version from the Releases section.

Otherwise, check doc/installation.md for instructions on how to build OpenPose from source.

Quick Start

Most users do not need the OpenPose C++/Python API, but can simply use the OpenPose Demo:

OpenPose Demo: To easily process images/video/webcam and display/save the results. See doc/demo_overview.md. E.g., run OpenPose in a video with:

# Ubuntu
./build/examples/openpose/openpose.bin --video examples/media/video.avi
:: Windows - Portable Demo
bin\OpenPoseDemo.exe --video examples\media\video.avi

Calibration toolbox: To easily calibrate your cameras for 3-D OpenPose or any other stereo vision task. See doc/modules/calibration_module.md.
OpenPose C++ API: If you want to read a specific input, and/or add your custom post-processing function, and/or implement your own display/saving, check the C++ API tutorial on examples/tutorial_api_cpp/ and doc/library_introduction.md. You can create your custom code on examples/user_code/ and quickly compile it with CMake when compiling the whole OpenPose project. Quickly add your custom code: See examples/user_code/README.md for further details.
OpenPose Python API: Analogously to the C++ API, find the tutorial for the Python API on examples/tutorial_api_python/.
Adding an extra module: Check doc/library_add_new_module.md.
Standalone face or hand detector:
- Face keypoint detection without body keypoint detection: If you want to speed it up (but also reduce amount of detected faces), check the OpenCV-face-detector approach in doc/standalone_face_or_hand_keypoint_detector.md.
- Use your own face/hand detector: You can use the hand and/or face keypoint detectors with your own face or hand detectors, rather than using the body detector. E.g., useful for camera views at which the hands are visible but not the body (OpenPose detector would fail). See doc/standalone_face_or_hand_keypoint_detector.md.

Output

Output (format, keypoint index ordering, etc.) in doc/output.md.

Speeding Up OpenPose and Benchmark

Check the OpenPose Benchmark as well as some hints to speed up and/or reduce the memory requirements for OpenPose on doc/speed_up_preserving_accuracy.md.

Foot Dataset

Check the foot dataset website and new OpenPose paper for more information.

Send Us Failure Cases and Feedback!

Our library is open source for research purposes, and we want to continuously improve it! So please, let us know if...

... you find videos or images where OpenPose does not seems to work well. Feel free to send them to openposecmu@gmail.com (email only for failure cases!), we will use them to improve the quality of the algorithm!
... you find any bug (in functionality or speed).
... you added some functionality to some class or some new Worker subclass which we might potentially incorporate.
... you know how to speed up or improve any part of the library.
... you have a request about possible functionality.
... etc.

Just comment on GitHub or make a pull request and we will answer as soon as possible! Send us an email if you use the library to make a cool demo or YouTube video!

Authors and Contributors

OpenPose is authored by Gines Hidalgo, Zhe Cao, Tomas Simon, Shih-En Wei, Hanbyul Joo, and Yaser Sheikh. Currently, it is being maintained by Gines Hidalgo and Yaadhav Raaj. The original CVPR 2017 repo includes Matlab and Python versions, as well as the training code. The body pose estimation work is based on the original ECCV 2016 demo.

In addition, OpenPose would not be possible without the CMU Panoptic Studio dataset.

We would also like to thank all the people who helped OpenPose in any way. The main contributors are listed in doc/contributors.md.

Citation

Please cite these papers in your publications if it helps your research (the face keypoint detector was trained using the procedure described in [Simon et al. 2017] for hands):

@inproceedings{cao2018openpose,
  author = {Zhe Cao and Gines Hidalgo and Tomas Simon and Shih-En Wei and Yaser Sheikh},
  booktitle = {arXiv preprint arXiv:1812.08008},
  title = {Open{P}ose: realtime multi-person 2{D} pose estimation using {P}art {A}ffinity {F}ields},
  year = {2018}
}

@inproceedings{cao2017realtime,
  author = {Zhe Cao and Tomas Simon and Shih-En Wei and Yaser Sheikh},
  booktitle = {CVPR},
  title = {Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields},
  year = {2017}
}

@inproceedings{simon2017hand,
  author = {Tomas Simon and Hanbyul Joo and Iain Matthews and Yaser Sheikh},
  booktitle = {CVPR},
  title = {Hand Keypoint Detection in Single Images using Multiview Bootstrapping},
  year = {2017}
}

@inproceedings{wei2016cpm,
  author = {Shih-En Wei and Varun Ramakrishna and Takeo Kanade and Yaser Sheikh},
  booktitle = {CVPR},
  title = {Convolutional pose machines},
  year = {2016}
}

Links to the papers:

License

OpenPose is freely available for free non-commercial use, and may be redistributed under these conditions. Please, see the license for further details. Interested in a commercial license? Check this FlintBox link. For commercial queries, use the Directly Contact Organization section from the FlintBox link and also send a copy of that message to Yaser Sheikh.

CMU-Perceptual-Computing-Lab/openpose