ComfyUI Extension: ComfyUI-YOLO

Authored by kadirnar

Created

Updated

70 stars

Nodes:Load Ultralytics Model, Ultralytics Inference, Ultralytics Visualization, Convert to Dictionary, BBox to XYWH

README

<div align="center"> <h2> ComfyUI-YOLO: Ultralytics-Powered Object Recognition for ComfyUI </h2> <div> <img width="800" alt="teaser" src="doc\yolo_logo.webp"> </div> </div>

🛠️ Installation

cd custom/nodes
git clone https://github.com/kadirnar/ComfyUI-YOLO

🤗 Model Hub

You can find the models on the Ultralytics Model Hub

cd models/ultralytics
wget https://github.com/ultralytics/assets/releases/download/v8.2.0/model_name.pt

Ultralytics Object Detection

<img width="1000" alt="Object Detection" src="doc\comfyui_object_det.png">

Ultralytics Video Object Detection

<img width="1000" alt="Object Detection" src="doc\comfyui_yolo_video.png">

Ultralytics Object Segmentation

<img width="1000" alt="Object Segmentation" src="doc\comfyui_object_seg.png">

Ultralytics Object Pose Estimation

<img width="1000" alt="object detection" src="doc\comfyui_object_pose.png">

Ultralytics Resize Image

<img width="1000" alt="object detection" src="doc/comfyui_resize.png">

Ultralytics Bbox Visualization

<img width="1000" alt="object detection" src="doc/comfyui_box_vis.png">