ComfyUI Extension: Comfyui-Yolov8-JSON

Authored by prodogape

Created

Updated

12 stars

This node is mainly based on the Yolov8 model for object detection, and it outputs related images, masks, and JSON information.

README

Comfyui-Yolov8-JSON

This node is mainly based on the Yolov8 model for object detection, and it outputs related images, masks, and JSON information.

image image

README

NODES

|name |description | |-----------------------------|----------------------------------------------------------------| |Load Yolov8 Model |Default, select the Yolov8 model, supports automatic download | |Load Yolov8 Model From Path |Load the model from the specified path | |Apply Yolov8 Model |Apply Yolov8 detection model | |Apply Yolov8 Model Seg |Apply Yolov8 segmentation model | |Save Labelme Json |Save the original image and the corresponding Labelme JSON format file to the output directory, with support for custom directories | |Draw Labelme Json |Based on the Labelme JSON, draw recognition boxes on the image and output masks, with support for specifying the labels to display, changing colors, and renaming labels |

INSTALL

If you need to display JSON formatted data or save it, you need to install the Comfyui-Toolbox node in advance.

This node calls the official Python package, and you also need to install the following dependencies:

pip install -r requirements.txt

MODEL

This node supports automatic model downloads. You can manually download the models to the specified folder models/yolov8 from Yolov8 as shown below:

ComfyUI
    models
        yolov8
            yolov8l.pt
            yolov8s.pt
            yolov8n.pt
            yolov8m.pt
            yolov8x.pt
            yolov8l-seg.pt
            yolov8s-seg.pt
            yolov8n-seg.pt
            yolov8m-seg.pt
            yolov8x-seg.pt