[doc] 📝 update dataset
This commit is contained in:
parent
584ba2c1e8
commit
be44c5b540
102
doc/dataset.md
Normal file
102
doc/dataset.md
Normal file
@ -0,0 +1,102 @@
|
||||
<!--
|
||||
* @Date: 2021-06-07 11:57:34
|
||||
* @Author: Qing Shuai
|
||||
* @LastEditors: Qing Shuai
|
||||
* @LastEditTime: 2021-06-07 14:45:17
|
||||
* @FilePath: /EasyMocapRelease/doc/dataset.md
|
||||
-->
|
||||
# EasyMoCap - Dataset
|
||||
|
||||
For convenience, all of the data used by EasyMoCap share the same format.
|
||||
|
||||
## Input structure
|
||||
|
||||
```bash
|
||||
<seq>
|
||||
├── intri.yml
|
||||
├── extri.yml
|
||||
└── videos
|
||||
├── 1.mp4
|
||||
├── 2.mp4
|
||||
├── ...
|
||||
├── 8.mp4
|
||||
└── 9.mp4
|
||||
```
|
||||
|
||||
You can use this commond to extract the videos to images:
|
||||
```bash
|
||||
python3 scripts/preprocess/extract_video.py ${data} --no2d
|
||||
```
|
||||
|
||||
After this, the folder will be like:
|
||||
```bash
|
||||
<seq>
|
||||
├── intri.yml
|
||||
├── extri.yml
|
||||
└── images
|
||||
├── 1
|
||||
│ ├── 000000.jpg
|
||||
│ ├── 000001.jpg
|
||||
│ ├── 000002.jpg
|
||||
│ └── ...
|
||||
├── 2
|
||||
│ ├── 000000.jpg
|
||||
│ ├── 000001.jpg
|
||||
│ ├── 000002.jpg
|
||||
│ └── ...
|
||||
├── ...
|
||||
├── ...
|
||||
├── 8
|
||||
│ ├── 000000.jpg
|
||||
│ ├── 000001.jpg
|
||||
│ ├── 000002.jpg
|
||||
│ └── ...
|
||||
└── 9
|
||||
├── 000000.jpg
|
||||
├── 000001.jpg
|
||||
├── 000002.jpg
|
||||
└── ...
|
||||
```
|
||||
|
||||
## 2D Pose
|
||||
|
||||
For each image, we record its 2D pose in a `json` file. For an image at `root/images/1/000000.jpg`, the 2D pose willl store at `root/annots/1/000000.json`. The content of the annotation file is:
|
||||
|
||||
```bash
|
||||
{
|
||||
"filename": "images/0/000000.jpg",
|
||||
"height": <the height of image>,
|
||||
"width": <the width of image>,
|
||||
"annots:[
|
||||
{
|
||||
'personID': 0, # ID of person
|
||||
'bbox': [l, t, r, b, conf],
|
||||
'keypoints': [[x0, y0, c0], [x1, y1, c1], ..., [xn, yn, cn]],
|
||||
'area': <the area of bbox>
|
||||
},
|
||||
{
|
||||
'personID': 1, # ID of person
|
||||
'bbox': [l, t, r, b, conf],
|
||||
'keypoints': [[x0, y0, c0], [x1, y1, c1], ..., [xn, yn, cn]],
|
||||
'area': <the area of bbox>
|
||||
}
|
||||
]
|
||||
}
|
||||
```
|
||||
|
||||
The definition of the `keypoints` is `body25`. If you want to use other definitions, you should add it to `easymocap/dataset/config.py`
|
||||
|
||||
## 3D Pose
|
||||
|
||||
```bash
|
||||
[
|
||||
{
|
||||
'id': <id>, # the person ID
|
||||
'keypoints3d': [[x0, y0, z0, c0], [x1, y1, z0, c1], ..., [xn, yn, zn, cn]], # x,y,z is the 3D coordinates, c means the confidence of this joint. If the c=0, it means this joint is invisible.
|
||||
},
|
||||
{
|
||||
'id': <id>, # the person ID
|
||||
'keypoints3d': [[x0, y0, z0, c0], [x1, y1, z0, c1], ..., [xn, yn, zn, cn]], # x,y,z is the 3D coordinates, c means the confidence of this joint. If the c=0, it means this joint is invisible.
|
||||
}
|
||||
]
|
||||
```
|
Loading…
Reference in New Issue
Block a user