Update: 24/01/2026 Qwen-VL Vision Model Ideal for Models like Flux_2, Qwen ... 13.08. Fixed a small Bug Update: 08/09/2025I had issues with the image loader in the old workflow, so I wrote a custom node myself that loads the images without errors.This time, it uses GPT for captioning along with a custom instruction.It can also be swapped out for open-source VLMs, but GPT works best overall. Update 24.10.24: Added Joytag Caption Update 29.04.24: I have changed the vision model from Moondream2 to llava.For the llava model to work, Ollama must be installed. This allows llava to run locally and comfyui to communicate with llava via a local API. <a target="_blank" rel="ugc" href="https://github.com/stavsap/comfyui-ollama">Ollama Github:</a> Update 25.03.24: The bug that was causing incorrect counting has been fixed.Comfy Moondream Caption (Dataset Caption Tool for Comfyui)Welcome to Comfy Moondream Caption, a tool designed to generate captions for images using a local LMM. Follow the steps below to get started:<img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/c125ff12-b935-4360-a12e-44fd4605b21b/width=525/c125ff12-b935-4360-a12e-44fd4605b21b.jpeg" />The workflow works for datasets with up to 9999 images.Select Image Folder:Navigate to the "folder_path" option and select the folder where your images are stored. Ensure that the images are in the supported format (e.g., PNG, JPEG).Naming Convention:Ensure that the images in the selected folder are sequentially numbered, such as "0001.png", "0002.png", and so on. This numbering is crucial for proper processing.Enable Batch Captioning:To enable batch captioning, go to the "Extra Options" section in the Comfy UI.Select "Auto Queue" to ensure that captions are generated for all images in the selected folder.<img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/c39eba83-f19a-4389-a302-3e16c46076a6/width=525/c39eba83-f19a-4389-a302-3e16c46076a6.jpeg" />Generate Captions:Once the folder and options are set, click on the "Que Prompt" button to initiate the caption generation process.Comfy Moondream Caption will analyze each image in the folder and generate captions for them.Post-Processing:After the caption generation process is complete, it's essential to reset the start index manually to 0. This ensures that the tool starts processing from the beginning for the next dataset.<img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/ad4302b3-21b4-47b4-b536-1ad5578f6240/width=525/ad4302b3-21b4-47b4-b536-1ad5578f6240.jpeg" /><img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/9af41296-840d-491e-8df8-dbc082e8cca0/width=525/9af41296-840d-491e-8df8-dbc082e8cca0.jpeg" />Repeat for Additional Datasets:If you have more datasets to analyze, repeat the process by selecting a new folder containing images.Installation:To set up this workflow, you'll need the ComfyUI Moondream custom nodes developed by Kijai. You can find them at: <a target="_blank" rel="ugc" href="https://github.com/kijai/ComfyUI-moondream">https://github.com/kijai/ComfyUI-moondream</a>

Dataset Caption Tool for Comfyui

datasetCaptionToolFor_ollamaLlavaVision.zip

Files with SHA256: 9e0c47afd1754a6f367d01231cc8db0d8b9c6ebc0204a05ef2285cece5168611