Views
No views yet
1from transformers import Qwen2_5_VLForConditionalGeneration, AutoProcessor
2from qwen_vl_utils import process_vision_info
3
4model = Qwen2_5_VLForConditionalGeneration.from_pretrained("kkk5/Re3Cap-CLIP-Qwen2.5-VL-7B", torch_dtype="auto", device_map="auto")
5processor = AutoProcessor.from_pretrained("kkk5/Re3Cap-CLIP-Qwen2.5-VL-7B")
6
7messages = [{"role": "user", "content": [
8 {"type": "image", "image": "your_image.jpg"},
9 {"type": "text", "text": "Caption this image as accurately as possible, without speculation. Describe what you see."},
10]}]
11
12text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
13image_inputs, video_inputs = process_vision_info(messages)
14inputs = processor(text=[text], images=image_inputs, videos=video_inputs, padding=True, return_tensors="pt").to(model.device)
15
16output_ids = model.generate(**inputs, max_new_tokens=1024)
17output_text = processor.batch_decode(output_ids[:, inputs.input_ids.shape[1]:], skip_special_tokens=True)[0]
18print(output_text)1@inproceedings{jia2026re3cap,
2 title = {Re$^3$Cap: Retrieval-Guided Refinement for Image Captioning Enhancement via Reinforcement Learning},
3 author = {Jia, Haonan and Dong, Shichao and Sun, Zenghui and Zheng, Jiawen and Miao, Ziqi and Shi, Gege and Zhao, Qiuyu and Lan, Jinsong and Zhu, Xiaoyong and Zheng, Bo},
4 booktitle = {Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing},
5 year = {2026}
6}