A tool for obscuring specified regions on an image.
"""
from functools import lru_cache
-from typing import Tuple, Type, List
+from typing import Tuple, Type, List, Optional
from PIL import Image, ImageFilter
@@ -425,7 +425,7 @@ Source code for imgutils.operate.censor_
[docs]def censor_nsfw(image: ImageTyping, method: str, nipple_f: bool = False, penis: bool = True, pussy: bool = True,
- level: str = 's', version: str = 'v1.0', max_infer_size=640,
+ level: str = 's', version: str = 'v1.0', model_name: Optional[str] = None,
conf_threshold: float = 0.3, iou_threshold: float = 0.7, **kwargs):
"""
Applies censoring to sensitive areas in NSFW images based on object detection.
@@ -454,8 +454,9 @@ Source code for imgutils.operate.censor_
:param version: The version of the NSFW object detection model. Default is ``v1.0``.
:type version: str
- :param max_infer_size: The maximum size for inference. Default is ``640``.
- :type max_infer_size: int
+ :param model_name: Optional custom model name. If not provided, it will be constructed
+ from the version and level.
+ :type model_name: Optional[str]
:param conf_threshold: The confidence threshold for object detection. Default is ``0.3``.
:type conf_threshold: float
@@ -490,7 +491,14 @@ Source code for imgutils.operate.censor_
"""
image = load_image(image, mode='RGB')
- areas = detect_censors(image, level, version, max_infer_size, conf_threshold, iou_threshold)
+ areas = detect_censors(
+ image=image,
+ level=level,
+ version=version,
+ model_name=model_name,
+ conf_threshold=conf_threshold,
+ iou_threshold=iou_threshold,
+ )
c_areas = []
for area, label, score in areas:
diff --git a/dev/yolo/_sources/information/environment.result.ipynb.txt b/dev/yolo/_sources/information/environment.result.ipynb.txt
index 33c02cd0699..2e3b9f7adcc 100644
--- a/dev/yolo/_sources/information/environment.result.ipynb.txt
+++ b/dev/yolo/_sources/information/environment.result.ipynb.txt
@@ -29,10 +29,10 @@
"execution_count": 1,
"metadata": {
"execution": {
- "iopub.execute_input": "2024-09-17T15:18:58.391390Z",
- "iopub.status.busy": "2024-09-17T15:18:58.390792Z",
- "iopub.status.idle": "2024-09-17T15:18:59.574299Z",
- "shell.execute_reply": "2024-09-17T15:18:59.573535Z"
+ "iopub.execute_input": "2024-09-17T15:49:45.151023Z",
+ "iopub.status.busy": "2024-09-17T15:49:45.150812Z",
+ "iopub.status.idle": "2024-09-17T15:49:46.335900Z",
+ "shell.execute_reply": "2024-09-17T15:49:46.335037Z"
},
"pycharm": {
"name": "#%%\n"
@@ -53,7 +53,7 @@
"text": [
"CPU Brand: AMD EPYC 7763 64-Core Processor\n",
"CPU Count: 4\n",
- "CPU Freq: 2925.0015000000003 MHz\n",
+ "CPU Freq: 3107.5155 MHz\n",
"Memory Size: 15.606 GiB\n",
"Has CUDA: No\n"
]
diff --git a/dev/yolo/api_doc/operate/censor.html b/dev/yolo/api_doc/operate/censor.html
index 29629d13658..23ff98201c1 100644
--- a/dev/yolo/api_doc/operate/censor.html
+++ b/dev/yolo/api_doc/operate/censor.html
@@ -206,7 +206,7 @@ censor_areas
-
-imgutils.operate.censor_.censor_nsfw(image: str | PathLike | bytes | bytearray | BinaryIO | Image, method: str, nipple_f: bool = False, penis: bool = True, pussy: bool = True, level: str = 's', version: str = 'v1.0', max_infer_size=640, conf_threshold: float = 0.3, iou_threshold: float = 0.7, **kwargs)[source]
+imgutils.operate.censor_.censor_nsfw(image: str | PathLike | bytes | bytearray | BinaryIO | Image, method: str, nipple_f: bool = False, penis: bool = True, pussy: bool = True, level: str = 's', version: str = 'v1.0', model_name: str | None = None, conf_threshold: float = 0.3, iou_threshold: float = 0.7, **kwargs)[source]
Applies censoring to sensitive areas in NSFW images based on object detection.
The censor area selected by this function is provided by the imgutils.detect.censor.detect_censors() function.
@@ -220,7 +220,8 @@ censor_nsfws (small), n (nano, faster than s). Default is s.
version (str) – The version of the NSFW object detection model. Default is v1.0.
max_infer_size (int) – The maximum size for inference. Default is 640.
model_name (Optional[str]) – Optional custom model name. If not provided, it will be constructed
+from the version and level.
conf_threshold (float) – The confidence threshold for object detection. Default is 0.3.
iou_threshold (float) – The IoU (Intersection over Union) threshold for non-maximum suppression. Default is 0.7.
kwargs – Additional keyword arguments to be passed to the censor method.
Run Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment InformationRun Environment Information