dev(narugo): add censor detect v0.9

app.py

@@ -91,7 +91,7 @@ if __name__ == '__main__':
                         gr_censor_infer_size = gr.Slider(480, 960, value=640, step=32, label='Max Infer Size')
                         with gr.Row():
                             gr_censor_iou_threshold = gr.Slider(0.0, 1.0, 0.5, label='IOU Threshold')
-                            gr_censor_score_threshold = gr.Slider(0.0, 1.0, 0.35, label='Score Threshold')
+                            gr_censor_score_threshold = gr.Slider(0.0, 1.0, 0.25, label='Score Threshold')
 
                         gr_censor_submit = gr.Button(value='Submit', variant='primary')
 

censor.py

@@ -9,6 +9,7 @@ from plot import detection_visualize
 from yolo_ import _image_preprocess, _data_postprocess
 
 _CENSOR_MODELS = [
+    'censor_detect_v0.9_s',
     'censor_detect_v0.8_s',
     'censor_detect_v0.7_s',
 ]
@@ -27,7 +28,7 @@ _LABELS = ['nipple_f', 'penis', 'pussy']
 
 
 def detect_censors(image: ImageTyping, model_name: str, max_infer_size=640,
-                   conf_threshold: float = 0.35, iou_threshold: float = 0.5) \
+                   conf_threshold: float = 0.25, iou_threshold: float = 0.5) \
         -> List[Tuple[Tuple[int, int, int, int], str, float]]:
     image = load_image(image, mode='RGB')
     new_image, old_size, new_size = _image_preprocess(image, max_infer_size)
@@ -38,6 +39,6 @@ def detect_censors(image: ImageTyping, model_name: str, max_infer_size=640,
 
 
 def _gr_detect_censors(image: ImageTyping, model_name: str, max_infer_size=640,
-                       conf_threshold: float = 0.35, iou_threshold: float = 0.5):
+                       conf_threshold: float = 0.25, iou_threshold: float = 0.5):
     ret = detect_censors(image, model_name, max_infer_size, conf_threshold, iou_threshold)
     return detection_visualize(image, ret, _LABELS)