Cogvlm Image Caption banner
fblissjr fblissjr

Cogvlm Image Caption

Development community

Description

Using CogVLM and CogAgent for image captioning

Installation

This entry records only its repository, not the path inside it, so there is no exact command to give. Open the source below and copy the folder into ~/.claude/skills/, or the file into ~/.claude/agents/.

README

cogvlm-image-caption

CLI (Command Line Interface) script using CogVLM and CogAgent for image captioning. This script iterates over a folder of images and creates a caption `.txt` file for each image found. It is modified from the example in the CogVLM repository (https://github.com/THUDM/CogVLM).

Usage

Example usage:

python caption_hf_cli.py --folder_path [path_to_images] --quant [quant_value] --bf16 --from_pretrained [model_path_or_repo_id]

You can find CogVLM models on Hugging Face here: https://huggingface.co/THUDM

Contributing

Feel free to fork and modify the script as needed for your use case.