Views
No views yet

1from utils.video import read_frames_decord
2from models.modeling_captioners import AutoCaptioner
3
4captioner = AutoCaptioner.from_pretrained('path/to/checkpoints/CaRe-7B-Stage-1')
5frames = read_frames_decord(video_path='assets/demo.mp4', num_frames=32)
6description = captioner.describe(frames.unsqueeze(0))
7print(description[0])