Unverified Commit afcb9843 authored by narugo1992's avatar narugo1992 Committed by GitHub
Browse files

Merge pull request #164 from deepghs/dev/yoloseg 

dev(narugo): add yolo-seg support
parents 8bdd4b2c 55b2f813

docs/source/api_doc/detect/similarity.rst

+20 −5
Original line number Diff line number Diff line
@@ -8,23 +8,38 @@ imgutils.detect.similarity 




calculate_iou

------------------------------------------

----------------------------------------------------------



.. autofunction:: calculate_iou







bboxes_similarity

------------------------------------------

----------------------------------------------------------



.. autofunction:: bboxes_similarity







detection_similarity

------------------------------------------

----------------------------------------------------------



.. autofunction:: detection_similarity





calculate_mask_iou

----------------------------------------------------------



.. autofunction:: calculate_mask_iou





masks_similarity

----------------------------------------------------------



.. autofunction:: masks_similarity





detection_with_mask_similarity

----------------------------------------------------------



.. autofunction:: detection_with_mask_similarity

docs/source/api_doc/generic/index.rst

+1 −0
Original line number Diff line number Diff line
@@ -16,3 +16,4 @@ imgutils.generic 
    multilabel_timm

    siglip

    yolo

    yoloseg

docs/source/api_doc/generic/yoloseg.rst

0 → 100644
+22 −0
Original line number Diff line number Diff line 
imgutils.generic.yoloseg

=======================================



.. currentmodule:: imgutils.generic.yoloseg



.. automodule:: imgutils.generic.yoloseg







YOLOSegmentationModel

----------------------------------------------------------



.. autoclass:: YOLOSegmentationModel

    :members: __init__,model_names, predict, clear, make_ui, launch_demo





yolo_seg_predict

----------------------------------------------------------



.. autofunction:: yolo_seg_predict

imgutils/detect/__init__.py

+2 −1
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ from .hand import detect_hands 
from .head import detect_heads

from .nudenet import detect_with_nudenet

from .person import detect_person

from .similarity import calculate_iou, bboxes_similarity, detection_similarity

from .similarity import (calculate_iou, bboxes_similarity, detection_similarity,

                         calculate_mask_iou, masks_similarity, detection_with_mask_similarity)

from .text import detect_text

from .visual import detection_visualize

imgutils/detect/base.py

+3 −0
Original line number Diff line number Diff line 
from typing import Tuple



import numpy as np



BBoxTyping = Tuple[float, float, float, float]

BBoxWithScoreAndLabel = Tuple[BBoxTyping, str, float]

MaskWithScoreAndLabel = Tuple[BBoxTyping, str, float, np.ndarray]