Phân loại video zero-shot

Mô hình đa phương thức với Hugging Face

James Chapman

Curriculum Manager, DataCamp

Phân loại video zero-shot

 

Tình huống kinh doanh:

  • Quảng cáo của đối thủ đang thúc đẩy doanh số
  • Quảng cáo của họ tích cực hơn không?

 

Quảng cáo kem đánh răng thập niên 1950

1 https://repository.duke.edu/dc/adviews/dmbb37204
Mô hình đa phương thức với Hugging Face

CLAP

  • Contrastive Language Audio Pretraining
  • Tương tự phương pháp huấn luyện CLIP
    • Bộ mã hóa văn bản và âm thanh tách biệt
  • 633k cặp khớp: âm thanh + mô tả
  • Tiền huấn luyện đối sánh làm thẳng hàng embedding âm thanh + văn bản

Sơ đồ CLAP cho thấy việc khớp embedding

1 https://arxiv.org/html/2211.06687v4
Mô hình đa phương thức với Hugging Face

Cách tiếp cận: ZSL đa phương thức

Học zero-shot với CLAP

Phóng to cách ghép trễ phần 1: CLAP

1 https://arxiv.org/html/2310.02298v3
Mô hình đa phương thức với Hugging Face

Cách tiếp cận: ZSL đa phương thức

Học zero-shot với CLAP

+ học zero-shot với CLIP

  • Lớp văn bản chung
  • Kết hợp xác suất từ CLIP và CLAP

Phóng to cách ghép trễ phần 2: CLIP+CLAP

Mô hình đa phương thức với Hugging Face

Video và âm thanh

Tách âm thanh và hình ảnh .mp4 với moviepy:

from moviepy.editor import VideoFileClip
from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip

ffmpeg_extract_subclip("advert.mp4", 0, 5, "advert_5s.mp4")


video = VideoFileClip("advert_5s.mp4")

audio = video.audio

audio.write_audiofile("advert_5s.mp3")
Mô hình đa phương thức với Hugging Face

Chuẩn bị âm thanh và video

from decord import VideoReader
from PIL import Image
video_reader = VideoReader(video_path)
video = video_reader.get_batch(range(20)).asnumpy()
video = video[:, :, :, ::-1]
video = [Image.fromarray(frame) for frame in video]


from datasets import Dataset, Audio
audio_dataset = Dataset.from_dict({"audio": [audio_path]}).cast_column("audio", Audio())
audio_sample = audio_dataset[0]["audio"]["array"]
Mô hình đa phương thức với Hugging Face

Dự đoán từ video

emotions = ["joy", "fear", "anger", "sadness", "disgust", "surprise", "neutral"]


image_class = pipeline(model="openai/clip-vit-large-patch14", 
                       task="zero-shot-image-classification")


predictions = image_classifier(video, candidate_labels=emotions)
scores = [
    {l['label']: l['score'] for l in prediction}
    for prediction in predictions
]


avg_image_scores = {emotion: sum([s[emotion] for s in scores])/len(scores) for emotion in emotions}
print(f"Average scores: {avg_image_scores}")

Average scores: {'joy': 0.10063326267991216, 'fear': 0.0868348691612482, ...}
Mô hình đa phương thức với Hugging Face

Dự đoán âm thanh và kết hợp

audio_class = pipeline(model="laion/clap-htsat-unfused", 
                       task="zero-shot-audio-classification")


audio_scores = audio_class(audio_sample, candidate_labels=emotions)

audio_scores = {l['label']: l['score'] for l in audio_scores}

multimodal_scores = {emotion: (avg_image_scores[emotion] + audio_scores[emotion])/2
                     for emotion in emotions}
print(f"Multimodal scores: {multimodal_scores}")

Multimodal scores: {'joy': 0.3109628591220826, 'fear': 0.09013736313208938, 
'anger': 0.011454355076421053, 'sadness': 0.06018101833760738, 'disgust': 0.07207315033301712, 
'surprise': 0.252118631079793, 'neutral': 0.2030726027674973}
Mô hình đa phương thức với Hugging Face

Ayo berlatih!

Mô hình đa phương thức với Hugging Face

Preparing Video For Download...