AlextheEngineer / Ego3DHands

This repository is for the newest Ego3DHand dataset.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Ego3DHands

Ego3DHands is a large-scale synthetic dataset for the task of two-hand 3D global pose estimation. It provides images and corresponding labels with the presence of two hands in egocentric view generated using Blender. This dataset can also be used for the task of hand segmentation, 2D, 3D canonical hand pose estimation. For hand tracking in dynamic sequences, we provide a dynamic version of Ego3DHands. This dataset is introduced by our paper Two-hand Global 3D Pose Estimation Using Monocular RGB.

Each instance provides the following data for both hands:

  • Hand image with transparent background
  • Hand image with background
  • Segmentation masks
    • 15 classes with the following labels:
      • 0: Background
      • 1: Arm (1~7 is for the left hand)
      • 2: Palm
      • 3: Thumb
      • 4: Index finger
      • 5: Middle finger
      • 6: Ring finger
      • 7: Pinky
      • 8~14: Repeats the order from class 1 to 8 for the right hand.
  • Depth image
    • Enocded as RGB images such that Depth_val = 1.0 * B_val + 0.01 * G_val + 0.0001 * R_val (cm). Background has value of 0s.
  • 2D joint locations
    • There are a total of 22 joints for each hand. The ndarray has shape of (2, 22, 2), where the first dimension is for the left and right hand, second dimension is for the 22 joints and third dimension is for row and column percentage value. Top left is (0.0, 0.0) and bottom right is (1.0, 1.0).
      • 0: Elbow (usually not used and can be ignored)
      • 1: Wrist
      • 2~5: Thumb
      • 6~9: Index finger
      • 10~13: Middle finger
      • 14~17: Ring finger
      • 18~21: Pinky
  • 3D global joint locations
    • The ndarray has shape of (2, 22, 3), where the first and second dimension is the same as 2D ndarray, the third dimension is for (row, col, depth). Row value increases going down, column increases going to the right and depth increases going away from the camera origin.
    • Normalized such that the bone length from wrist to the middle metacarpophalangeal joint (mMCP/root joint) has length of 10.0cm
  • 3D canonical joint locations
    • 3D global joint locations spherically rotated to center the mMCP, zero-centered on mMCP and normalized so the bone formed by the wrist joint and mMCP (key bone) has length of 1.0).
    • Unlike common methods that generate the canonical poses by simply translating the absolute root joint to the origin, we spherically rotate the joint locations to align the root joint to the center of the view so the visual representation of the hand poses are consistent with the 3D canonical joint locations. In other words, hand poses that have the same visual representation at different global locations should have the same 3D canonical joint loations.
  • Camera Intrinsic Matrix
    • Since we provide our 2D coordinates in form of percentages of the image size, the actual 2D coordiantes can be easily computed for images with any resized dimensions. As a result, the camera intrisic matrix changes for different image sizes and can be computed as shown below.
      • For K = [[fx, 0, px], [0, fy, py], [0, 0, 1]]. Note that our coordinates are in the format of (row, column, depth), so x, y, z correlates with the row, column, depth dimension respectively.
      • For any joint in the dataset, px = IMG_H / 2, py = IMG_W / 2, fx = (x_2d - 0.5) * IMG_H * z_3d / x_3d and fy = (y_2d - 0.5) * IMG_W * z_3d / y_3d, where x_2d, x_3d, y_2d, y_3d, z_3d are the provided original values in the dataset for any joint.
      • For the image size (270, 480) we use in our paper, K = [[187.932, 0, 135], [0, 187.932, 240], [0, 0, 1]]. For any image with size (IMG_H, IMG_W), you could also compute the K matrix value using px = IMG_H / 2, py = IMG_W / 2, fx = 187.932 * IMG_H / 270 and fy = 187.932 * IMG_W / 480.
      • The K matrix is consistent for the entire dataset given a specific image size.

Note that some instances only have a single hand present, in that case the missing hand will have values of zero for its joint locations.

Ego3DHands static & dynamic

Ego3DHands dataset provides 2 different versions for the task of static and dynamic pose estiamtion respectively. The static version includes 50,000 training instances and 5,000 test instances. The background images for the static version are randomly selected from approximately 20,000 images within 100 different scene categories from online sources. Additionally, the background images are randomly flipped horizontally and color augmented. The dynamic version includes 100 training videos and 10 test videos with 500 frames per video sequence. Each sequence has a unique background sequence selected from www.pexels.com.

Download

Please use the following links for downloading the datasets:

Ego3DHands (static): https://app.box.com/s/j5a27ilrxraz94ujlvg3gzf715fiq8ih

Ego3Dhands (dynamic): https://app.box.com/s/wtib1zbtmzu9wbwbw1miw8q1oivdbyx6

Evaluation

For the task of global hand pose estimation, we evaluate in terms of both the AUC for the PCK of the 3D canonical hand poses (pose accuracy) and a new metric of the AUC of the spherical PCK of the root joint that computes the distance accuracy and directional accuracy. Please see our paper for more details.

Visualization & Global Projection Algorithm

In order to visualize the 3D joints for both hands, you need to first create the proper environment in Anaconda. We ran the following commands to set up the environment on a Windows machine:

  • conda create --name ego3dhand_vis python=3.6
  • conda activate ego3dhand_vis
  • conda install -c anaconda numpy
  • conda install -c conda-forge opencv
  • conda install -c anaconda pillow
  • pip install pygame
  • conda install -c anaconda pyopengl

After setting up the environment, you can run the following command:

  • python visualize_3d_pose.py ego_3d_hand

This should load the 2D, 3D canonical and 3D global joint locations from a sample instance in Ego3DHands and output the visualization in predefined angles in "outputs/" directory. You should obtain the following output images:

We have included camera intrinsics for Stereo Tracking Benchmark Dataset (STB) and Rendered Hand Pose Dataset (RHP). However, you would need to add in the samples from these other datasets and process the joints the same way we do in order for the script to function properly. Samples from other datasets can be added as well but the corresponding camera intrinsics would also need to be added into the existing script (e.g. refer to the issue tab for running the algorithm on InterHand2.6M).

This script also contains our global projection algorithm, which computes the 3D global joint locations using the 2D, 3D canonical joint locations, camera intrinsics and key bone length. This algorithm can be applied to compute the global 3D keypoint locations of any object given the aforementioned information.

License

This dataset can only be used for scientific/non-commercial purposes. If you use this dataset in your research, please cite the corresponding paper

About

This repository is for the newest Ego3DHand dataset.


Languages

Language:Python 100.0%