Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

lint set minimum to py310 and update rules

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-10-29 11:28:09 -04:00
parent 567b9e7014
commit d43091f1fa
6 changed files with 12 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/nudenet/nudenet.py
+2 -2
View File
@@ -150,8 +150,8 @@ class NudeDetector:
return background
if fg_channels < 4: # make sure that overlay is rgba
log.warning('NudeNet overlay image does not have alpha channel')
foreground = cv2.cvtColor(foreground, cv2.COLOR_RGB2RGBA)
foreground[:, :, 3] = cv2.cvtColor(foreground, cv2.COLOR_BGR2GRAY)
foreground_color = cv2.cvtColor(foreground, cv2.COLOR_RGB2RGBA)
foreground[:, :, 3] = cv2.cvtColor(foreground_color, cv2.COLOR_BGR2GRAY)
fg_h, fg_w, fg_channels = foreground.shape
if x_offset is None: # center by default
x_offset = (bg_w - fg_w) // 2