dev(narugo): 1 more

app2.py

@@ -2,7 +2,7 @@ import os
 
 import gradio as gr
 
-from detection import EyesDetection, FaceDetection, HeadDetection, PersonDetection, HandDetection
+from detection import EyesDetection, FaceDetection, HeadDetection, PersonDetection, HandDetection, CensorDetection
 
 _GLOBAL_CSS = """
 .limit-height {
@@ -26,7 +26,7 @@ if __name__ == '__main__':
             with gr.Tab('Hand Detection'):
                 HandDetection().make_ui()
             with gr.Tab('Censor Point Detection'):
-                pass
+                CensorDetection().make_ui()
             with gr.Tab('Manbits Detection\n(Deprecated)'):
                 pass
 

detection/__init__.py

@@ -1,4 +1,5 @@
 from .base import ObjectDetection, DeepGHSObjectDetection
+from .censor import CensorDetection
 from .eyes import EyesDetection
 from .face import FaceDetection
 from .hand import HandDetection

detection/censor.py

@@ -0,0 +1,33 @@
+import re
+from typing import List, Tuple
+
+from imgutils.data import ImageTyping
+from imgutils.detect.censor import detect_censors, _LABELS
+
+from .base import DeepGHSObjectDetection
+
+
+def _parse_model_name(model_name: str):
+    matching = re.fullmatch(r'^censor_detect_(?P<version>[\s\S]+?)_(?P<level>[\s\S]+?)$', model_name)
+    return matching.group('version'), matching.group('level')
+
+
+class CensorDetection(DeepGHSObjectDetection):
+    def __init__(self):
+        DeepGHSObjectDetection.__init__(self, repo_id='deepghs/anime_censor_detection')
+
+    def _get_default_model(self) -> str:
+        return 'censor_detect_v1.0_s'
+
+    def _get_default_iou_and_score(self, model_name: str) -> Tuple[float, float]:
+        return 0.7, 0.3
+
+    def _get_labels(self, model_name: str) -> List[str]:
+        return _LABELS
+
+    def detect(self, image: ImageTyping, model_name: str,
+               iou_threshold: float = 0.7, score_threshold: float = 0.25) \
+            -> List[Tuple[Tuple[float, float, float, float], str, float]]:
+        version, level = _parse_model_name(model_name)
+        return detect_censors(image, level=level, version=version,
+                              conf_threshold=score_threshold, iou_threshold=iou_threshold)