ComfyUI-OmDet

★ 4

目标检测掩码生成Labelme标注模型自动下载

基于OmDet的目标检测节点，自动下载模型，输出检测图、实例掩码与Labelme JSON，便于标注与后处理

💡 对图片批量检测并导出掩码与Labelme标注文件

🍴 2 Forks💻 Python🔄 2024-06-14

📦

网盘下载

复制链接后前往夸克网盘下载

https://pan.quark.cn/s/9671236b7e59

📦 requirements.txt

lmdb
pytorch
torchvision
pytorch-cuda
pytorch
detectron2

📄 README

This node is mainly based on OmDet for object detection, and it outputs related images, masks, and Labelme JSON information.

README

|name |description |

|——————————|—————————————————|

|Apply OmDet |Apply OmDer model, supports automatic download |

you need to install the following dependencies:

pip install -r requirements.txt

This node supports automatic model downloads.

ComfyUI
    models
        OmDet
            OmDet-Turbo_tiny_SWIN_T.pth
            ViT-B-16.pt