nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

add tagging

Browse source
This commit is contained in:
Tom Young 2024-06-17 14:03:54 +00:00
parent 7fd26ba54f
commit b59f5892dd
1 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
docs/data_processing.md
View file

@ -65,8 +65,13 @@ python -m tools.datasets.datautil \
--refine-llm-caption \
--remove-empty-caption \
--output ${ROOT_META}/meta_clips_caption_cleaned.csv
# 4.4 Optionally generate tags (e.g., objects) based on the captions. This should output your_output_prefix_{key}.csv
torchrun --nproc_per_node 8 --standalone -m tools.caption.caption_llama3 ${ROOT_META}/meta_clips_caption_cleaned.csv --key objects --output_prefix your_output_prefix
```
For more information, please refer to:
- [Dataset Management](../tools/datasets/README.md)
- [Scene Detection and Video Splitting](../tools/scene_cut/README.md)