Add mlops/computer-vision-audio-models

This commit is contained in:
2026-07-10 16:11:12 +08:00
parent d503ec9baa
commit a7dc9b4b55
@@ -0,0 +1,85 @@
---
name: computer-vision-audio-models
description: "Specialized AI models: image segmentation (SAM / Segment Anything) and audio generation (AudioCraft / MusicGen)."
version: 1.0.0
author: Hermes Agent
license: MIT
metadata:
hermes:
tags: [computer-vision, audio-generation, sam, segment-anything, audiocraft, musicgen, audiogen, image-segmentation]
related_skills: [comfyui, llm-inference]
---
# Computer Vision & Audio Models
Two specialized model families for non-text AI tasks: zero-shot image segmentation and text-to-audio generation.
## Section 1: Segment Anything Model (SAM) — Zero-Shot Image Segmentation
Meta's Segment Anything Model for zero-shot image segmentation via points, boxes, or masks as prompts.
**When to use:** Object segmentation, image editing, data annotation, medical imaging, removing/replacing objects in images.
**Key features:**
- Zero-shot segmentation — no training data needed for new objects
- Point, box, and mask prompt types
- SAM 2 supports video segmentation with memory
- Automatic mask generation (segment everything)
- ONNX export for production deployment
**Quick start:**
```python
from segment_anything import SamPredictor, sam_model_registry
sam = sam_model_registry["vit_h"](checkpoint="sam_vit_h_4b8939.pth")
predictor = SamPredictor(sam)
predictor.set_image(image)
masks, scores, logits = predictor.predict(point_coords=input_point, point_labels=input_label)
```
**Pitfalls:**
- ViT-H checkpoint is ~2.4GB; use ViT-B (~375MB) for limited VRAM
- Point prompts need `point_labels` (1=foreground, 0=background)
- For SAM 2 video segmentation, frames must be processed sequentially
- Automatic mask generation is slow on CPU — use GPU
**See:** `references/segment-anything.md` for full API, advanced usage, and troubleshooting.
## Section 2: AudioCraft — Text-to-Audio Generation
Meta's AudioCraft for MusicGen (text-to-music) and AudioGen (text-to-sound effects).
**When to use:** Generating music from text descriptions, creating sound effects, audio for games/video, ambient soundscapes.
**Key features:**
- MusicGen: text-to-music with melody conditioning
- AudioGen: text-to-sound-effects
- Multiple model sizes (small/medium/large/melody)
- Streaming generation for real-time preview
- Fine-tuning support on custom audio data
**Quick start (MusicGen):**
```python
from audiocraft.models import MusicGen
from audiocraft.data.audio import audio_write
model = MusicGen.get_pretrained('medium')
model.set_generation_params(duration=8)
wav = model.generate(['happy rock song with guitar solo'])
audio_write('output', wav[0].cpu(), model.sample_rate, strategy="loudness")
```
**Quick start (AudioGen):**
```python
from audiocraft.models import AudioGen
model = AudioGen.get_pretrained('medium')
model.set_generation_params(duration=5)
wav = model.generate(['dog barking in the distance'])
```
**Pitfalls:**
- Large model requires ~16GB VRAM for generation; medium works on ~8GB
- `loudness` normalization strategy avoids clipping
- Melody conditioning requires a reference audio file (24kHz mono)
- Fine-tuning needs hours of clean audio data + significant GPU time
**See:** `references/audiocraft.md` for full API, advanced usage, and troubleshooting.