This Python tool is designed to generate captions for a set of images, utilizing the advanced capabilities of OpenAI's GPT-4 Vision API. It can handle image collections either from a ZIP file or a directory. The tool offers flexibility in captioning, providing options to describe images directly or in a creative style, like "in the style of Family Guy."
- Processes images from ZIP files or directories.
- Utilizes OpenAI's GPT-4 Vision API for caption generation.
- Supports custom caption formats, including style-based descriptions (e.g., "in the style of TOK").
- Captions are conveniently saved in a CSV file for easy access and reference.
- Python 3
- An active OpenAI API key with GPT-4 Vision API access.
- Necessary Python libraries:
requests,imgcat.
-
Clone the GitHub repository:
git clone https://github.com/ghostofpokemon/oCaption.git -
Change to the tool's directory:
cd oCaption -
Install the required Python dependencies:
pip install requests imgcat
Set your OpenAI API key as an environment variable:
export OPENAI_API_KEY="Your-API-Key-Here"
Launch the tool with Python:
python3 oCaption.py
Follow the prompts to input the path to your image folder or ZIP file, specify the TOK value (such as "TOK" or "Family Guy"), and choose a caption prefix if needed.
The tool generates detailed captions for each processed image, saving them in a caption.csv file located in the current working directory. The captions follow the format "a photo of [subject]" or "in the style of [TOK]" based on your preference.
We welcome contributions! To propose changes, please use the standard GitHub pull request process.
This tool is distributed under the MIT License.
For queries or feedback, feel free to open an issue in the GitHub repository.