AudioSetCaps
收藏资源简介:
AudioSetCaps是一个通过使用大型音频和语言模型自动生成管道来丰富音频描述数据集。该数据集包含音频文件及其对应的描述标签。
AudioSetCaps is an audio description dataset enriched through an automated generation pipeline leveraging large-scale audio and language models. This dataset contains audio files along with their corresponding descriptive labels.
AudioSetCaps: Enriched Audio Captioning Dataset Generation Using Large Audio Language Models
数据集准备
- 需要准备一个包含文件名和标签的CSV文件。
运行代码步骤
-
创建Qwen环境 bash conda create -n qwen python=3.10 pip install -r qwen_requirement.txt conda install FFmpeg
-
安装Flash Attention
- 链接: flash attention
-
创建CLAP环境 bash conda create -n clap python=3.10 pip install laion-clap git clone https://github.com/LAION-AI/CLAP.git cd CLAP pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html pip install -r requirements.txt
-
安装Ollama bash curl https://ollama.ai/install.sh | sh
-
修改conda.sh路径
- 在
Audiosetcaps_qwen_main.sh和Audiosetcaps_Mistral_main.sh中修改conda.sh路径。
- 在
-
生成Qwen-audio描述 bash bash Audiosetcaps_qwen_main.sh
-
启动Ollama服务 bash ollama serve
-
运行Mistral bash ollama run mistral
-
下载CLAP模型
-
配置路径和参数
- 在
config.py中设置路径和参数。
- 在
-
生成Mistral描述 bash bash Audiosetcaps_Mistral_main.sh




