Skip to content

Commit 2f5f771

Browse files
committed
Update README.md to clarify pre-trained model setup, highlighting automatic downloads from Hugging Face and providing alternative options for local weights.
1 parent 11ab3d3 commit 2f5f771

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

README.md

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,9 @@ pip install "fmpose3d[animals]"
5151

5252
This visualization script is designed for single-frame based model, allowing you to easily run 3D human pose estimation on any single image.
5353

54-
Before testing, make sure you have the pre-trained model ready.
55-
You may either use the model trained by your own or download ours from [here](https://drive.google.com/drive/folders/1235_UgUQXYZtjprBOv2ZJJHY2KOAS_6p?usp=sharing) and place it in the `./pre_trained_models` directory.
54+
Pre-trained weights are downloaded automatically from [Hugging Face](https://huggingface.co/MLAdaptiveIntelligence/FMPose3D) the first time you run inference, so no manual setup is needed.
55+
56+
Alternatively, you can use your own trained weights or download ours from [Google Drive](https://drive.google.com/drive/folders/1aRZ6t_6IxSfM1nCTFOUXcYVaOk-5koGA?usp=sharing), place them in the `./pre_trained_models` directory, and set `model_weights_path` in the shell script (e.g. `demo/vis_in_the_wild.sh`).
5657

5758
Next, put your test images into folder `demo/images`. Then run the visualization script:
5859
```bash
@@ -93,7 +94,7 @@ sh ./scripts/FMPose3D_train.sh
9394

9495
### Inference
9596

96-
First, download the folder with pre-trained model from [here](https://drive.google.com/drive/folders/1235_UgUQXYZtjprBOv2ZJJHY2KOAS_6p?usp=sharing) and place it in the './pre_trained_models' directory.
97+
Pre-trained weights are fetched automatically from Hugging Face on the first run. You can also use local weights by setting `model_weights_path` in the shell script (see [Demos](#testing-on-in-the-wild-images-humans) above for details).
9798

9899
To run inference on Human3.6M:
99100

0 commit comments

Comments
 (0)