Skip to main content

ZZ-Image-Caption

Image captioner CLI using BLIP and BLIP2 models

Installation

Requirements:

  • Python 3.10 or higher

Install using pip:

pip install zz-image-caption

Install pytorch

You may need to install pytorch separately depending on your system to use CUDA (default to use CPU if not available).

Usage

Basic usage:

Print caption for an image to the console

caption image.jpg

Advanced usage:

Rename images in a directory with their captions

caption images/ -o filename

Write metadata for images in a directory with their captions

caption images/ -o metadata

Print caption for an image to the console using the BLIP2 model

caption image.jpg --blip2

Command Line Interface Options

The following table lists all the command-line arguments available with descriptions and additional details:

Argument Type Choices Default Description
-v, --version flag Display the version of the tool.
input string Path to the input image file or directory.
-o, --output string text, json, metadata, filename Specify the output type.
-a, --append string Append string to caption output.
-t, --token integer 32 Max token length for captioning.
-b, --batch integer 1 Batch size for captioning.
-p, --prompt string Prompt for captioning.
--temp, --temperature float 1.0 Temperature for captioning.
--seed integer Seed for reproducibility.
--large flag Use the large model for captioning.
--cpu flag Use CPU instead of GPU (not recommended).
--blip2 flag Use Blip2 model for captioning.
--verbose flag Print verbose output.
--debug flag Print debug output.

Help:

caption --help

Release files for zz-img-caption 0.0.3

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for zz-img-caption 0.0.3
File Size Uploaded
zz-img-caption-0.0.3.tar.gz 7.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for zz-img-caption 0.0.3
File Interpreter ABI Platform
zz_img_caption-0.0.3-py3-none-any.whl Python 3 none any Details

Total release size: 14.6 kB

Release files / zz-img-caption-0.0.3.tar.gz

Download URL zz-img-caption-0.0.3.tar.gz
Size 7.1 kB
Tags Source
SHA-256 checksum
How to use checksums
1b624642758fa555feab16cd9ddda51e1474491c65fe3a25ed26e58d90a74afe
BLAKE2b-256 checksum
How to use checksums
0bbe35134e3ab65e8ac84f7580ea55ac4d01ddd52cbeab9b6f0083df8b68f6cf
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.0.0 CPython/3.10.11

Release files / zz_img_caption-0.0.3-py3-none-any.whl

Download URL zz_img_caption-0.0.3-py3-none-any.whl
Size 7.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
04ea82f1019f69602491534d00397af2248bb6fd53beb5fa818903bc40ddeff5
BLAKE2b-256 checksum
How to use checksums
d91c90e16fde97d08210bbca53c6422bb9704bfe7884fbca76de52fba9adccea
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/5.0.0 CPython/3.10.11

Release history Release notifications | RSS feed

This release

0.0.3 This release

2 release files

0.0.2

2 release files

0.0.1

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page