Skip to content

Commit

Permalink
add docs
Browse files Browse the repository at this point in the history
  • Loading branch information
SangbumChoi committed Aug 20, 2024
1 parent f064009 commit e9c6b1e
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion docs/source/en/model_doc/vitpose.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,10 @@ The original code can be found [here](https://github.com/ViTAE-Transformer/ViTPo

## Usage Tips

The current model utilizes a 2-step inference pipeline. The first step involves placing a bounding box around the region corresponding to the person. After that, the second step uses ViTPose to predict the keypoints.
- To enable MoE(Mixture or Expert) function in backbone, user have to give appropriate input indices into the backbone model.
However, it is not used in default parameters.
- The current model utilizes a 2-step inference pipeline. The first step involves placing a bounding box around the region corresponding to the person.
After that, the second step uses ViTPose to predict the keypoints.

```py
>>> import torch
Expand Down

0 comments on commit e9c6b1e

Please sign in to comment.