ComfyUI Extension: Comfyui-Yolov8-JSON

Authored by prodogape

Created

Updated

20 stars

This node is mainly based on the Yolov8 model for object detection, and it outputs related images, masks, and JSON information.[w/Repository url is changed. Please remove previous one and reinstall.]

Custom Nodes (0)

    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