Filmtitle Beit GPT2

🚀 图像标题生成模型

本模型用于为任意电影海报生成中文标题，结合了先进的图像特征提取和文本生成技术，为电影海报的内容理解和标题创作提供了高效的解决方案。

🚀 快速开始

模型描述

该模型用于生成随机电影海报的中文标题。它基于 BEiT 和 GPT2 构建。

训练数据

训练数据包含 5043 张电影海报及其对应的中文标题，这些数据是从 Movie-Title-Post 收集而来。

如何使用

from transformers import VisionEncoderDecoderModel, ViTFeatureExtractor, AutoTokenizer
from PIL import Image

pretrained = "snzhang/FilmTitle-Beit-GPT2"
model = VisionEncoderDecoderModel.from_pretrained(pretrained)
feature_extractor = ViTFeatureExtractor.from_pretrained(pretrained)
tokenizer = AutoTokenizer.from_pretrained(pretrained)

image_path = "your image path"
image = Image.open(image_path)
if image.mode != "RGB":
        image = image.convert("RGB")
pixel_values = feature_extractor(images=image, return_tensors="pt").pixel_values

output_ids = model.generate(pixel_values, **gen_kwargs)
preds = tokenizer.batch_decode(output_ids, skip_special_tokens=True)
preds = [pred.strip() for pred in preds]
print(preds)

📚 详细文档

你可以在 FilmTitle-Beit-GPT2 中获取更多训练细节。

📄 许可证

本项目采用 Apache-2.0 许可证。