Update README.md

This commit is contained in:
Yuze He 2025-05-11 22:40:41 +08:00 committed by GitHub
parent ad087b855f
commit 951f76edfd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -55,7 +55,23 @@ Then download data and pretrained weights:
3. **Demo and test data**:
Download from this [Google Drive link](https://drive.google.com/file/d/1FZZjk0OvzETD5j4OODEghS_YppcS_ZbM/view?usp=sharing), then decompress the files into `./data/`.
Download from this [Google Drive link](https://drive.google.com/file/d/1FZZjk0OvzETD5j4OODEghS_YppcS_ZbM/view?usp=sharing), then decompress the files into `./data/`; or you can download from our 🤗 Hugging Face datasets library:
```python
from huggingface_hub import hf_hub_download, list_repo_files
# Get list of all files in repo
files = list_repo_files(repo_id="hyz317/PrimitiveAnything", repo_type="dataset")
# Download each file
for file in files:
file_path = hf_hub_download(
repo_id="hyz317/PrimitiveAnything",
filename=file,
repo_type="dataset",
local_dir='./data'
)
```
After downloading and organizing the files, your project directory should look like this: