Skip to content

Image captioning GUI using Florence-2.

License

Notifications You must be signed in to change notification settings

Zuellni/Qt-Caption

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

23 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Qt Caption

A simple image captioning GUI using Florence-2.

Installation

Create a new environment with mamba:

mamba create -n qt-caption git python pytorch pytorch-cuda torchvision -c conda-forge -c nvidia -c pytorch
mamba activate qt-caption

Clone the repository and install requirements:

git clone https://github.com/zuellni/qt-caption
pip install -r qt-caption/requirements.txt

Usage

Start the GUI with:

python qt-caption

Or with a script like this to hide the console:

@echo off
call mamba activate qt-caption
start pythonw qt-caption

Preview

Preview