鉴黄师,python也可以做!

640?wx_fmt=gif

相信大家都对鉴黄师这个职业很是羡慕,但是又害怕会带来身体负担,尤其是肾的负担,这次我们就来看看如何使用python鉴黄。

640?wx_fmt=jpeg

1、安装需要的库:

pip install Pillow

2、了解原理:

本程序根据颜色(肤色)找出图片中皮肤的区域,然后通过一些条件判断是否为色情图片

程序的关键步骤如下

  1. 遍历每个像素,检测像素颜色是否为肤色

  2. 将相邻的肤色像素归为一个皮肤区域,得到若干个皮肤区域

  3. 剔除像素数量极少的皮肤区域

我们定义非色情图片的判定规则如下

    (满足任意一个判定为真):

  1. 皮肤区域的个数小于 3 个

  2. 皮肤区域的像素与图像所有像素的比值小于 15%

  3. 最大皮肤区域小于总皮肤面积的 45%

  4. 皮肤区域数量超过60个

这些规则你可以尝试更改,直到程序效果让你满意为止

关于像素肤色判定这方面,公式可以在网上找到很多,但世界上不可能有正确率 100% 的公式

你可以用自己找到的公式,在程序完成后慢慢调试

  • RGB 颜色模式

    第一种:r > 95 and g > 40 and g < 100 and b > 20 and max([r, g, b]) - min([r, g, b]) > 15 and abs(r - g) > 15 and r > g and r > b

    第二种:nr = r / (r + g + b), ng = g / (r + g + b), nb = b / (r +g + b)nr / ng > 1.185 and r * b / (r + g + b) ** 2 > 0.107 and r * g / (r + g + b) ** 2 > 0.112

  • HSV 颜色模式

    h > 0 and h < 35 and s > 0.23 and s < 0.68

  • YCbCr 颜色模式

    97.5 <= cb <= 142.5 and 134 <= cr <= 176

一幅图像有零个到多个的皮肤区域,程序按发现顺序给它们编号,第一个发现的区域编号为 0,第 n 个发现的区域编号为 n-1

我们用一种类型来表示像素,我们给这个类型取名为 Skin,包含了像素的一些信息:唯一的 编号(id),是/否肤色(skin),皮肤区域号(region),横坐标(x),纵坐标(y

遍历所有像素时,我们为每个像素创建一个与之对应的 Skin 对象,并设置对象的所有属性

其中 region 属性即为像素所在的皮肤区域编号,创建对象时初始化为无意义的 None

关于每个像素的 id 值,左上角为原点,像素的 id 值按像素坐标排布,那么看起来如下图

640?wx_fmt=png

其实 id 的顺序也即遍历的顺序

遍历所有像素时,创建 Skin 对象后,如果当前像素为肤色,且相邻的像素有肤色的,那么我们把这些肤色像素归到一个皮肤区域

相邻像素的定义:通常都能想到是当前像素周围的 8 个像素,然而实际上只需要定义 4 个就可以了,位置分别在当前像素的左方,左上方,正上方,右上方;因为另外四个像素都在当前像素后面,我们还未给这4个像素创建对应的 Skin 对象

640?wx_fmt=png

接下来实现细节部分

3、敲代码

导入所需要的模块


   
  1. import sys
  2. import os
  3. import _io
  4. from collections import namedtuple
  5. from PIL import Image

我们将设计一个 Nude 类:

class Nude(object):
  

这个类里面我们首先使用 collections.namedtuple() 定义一个 Skin 类型

Skin = namedtuple("Skin", "id skin region x y")
  

collections.namedtuple() 函数实际上是一个返回Python中标准元组类型子类的一个工厂方法。 你需要传递一个类型名和你需要的字段给它,然后它就会返回一个类,你可以初始化这个类,为你定义的字段传递值等。 详情参见官方文档。

然后定义 Nude 类的初始化方法


   
  1. def __init__(self, path_or_image):
  2.    # 若 path_or_image 为 Image.Image 类型的实例,直接赋值
  3.    if isinstance(path_or_image, Image.Image):
  4.        self.image = path_or_image    # 若 path_or_image 为 str 类型的实例,打开图片
  5.    elif isinstance(path_or_image, str):
  6.        self.image = Image.open(path_or_image)    # 获得图片所有颜色通道
  7.    bands = self.image.getbands()    # 判断是否为单通道图片(也即灰度图),是则将灰度图转换为 RGB 图
  8.    if len(bands) == 1:        # 新建相同大小的 RGB 图像
  9.        new_img = Image.new("RGB", self.image.size)        # 拷贝灰度图 self.image 到 RGB图 new_img.paste (PIL 自动进行颜色通道转换)
  10.        new_img.paste(self.image)
  11.        f = self.image.filename        # 替换 self.image
  12.        self.image = new_img
  13.        self.image.filename = f    # 存储对应图像所有像素的全部 Skin 对象
  14.    self.skin_map = []    # 检测到的皮肤区域,元素的索引即为皮肤区域号,元素都是包含一些 Skin 对象的列表
  15.    self.detected_regions = []    # 元素都是包含一些 int 对象(区域号)的列表
  16.    # 这些元素中的区域号代表的区域都是待合并的区域
  17.    self.merge_regions = []    # 整合后的皮肤区域,元素的索引即为皮肤区域号,元素都是包含一些 Skin 对象的列表
  18.    self.skin_regions = []    # 最近合并的两个皮肤区域的区域号,初始化为 -1
  19.    self.last_from, self.last_to = -1, -1
  20.    # 色情图像判断结果
  21.    self.result = None
  22.    # 处理得到的信息
  23.    self.message = None
  24.    # 图像宽高
  25.    self.width, self.height = self.image.size    # 图像总像素
  26.    self.total_pixels = self.width * self.height

注:

  • isinstane(object, classinfo) 如果参数 object 是参数 classinfo 的实例,返回真,否则假;参数 classinfo 可以是一个包含若干 type 对象的元祖,如果参数 object 是其中任意一个类型的实例,返回真,否则假

图片缩小


   
  1. def resize(self, maxwidth=1000, maxheight=1000):
  2.    """
  3.    基于最大宽高按比例重设图片大小,
  4.    注意:这可能影响检测算法的结果
  5.    如果没有变化返回 0
  6.    原宽度大于 maxwidth 返回 1
  7.    原高度大于 maxheight 返回 2
  8.    原宽高大于 maxwidth, maxheight 返回 3
  9.    maxwidth - 图片最大宽度
  10.    maxheight - 图片最大高度
  11.    传递参数时都可以设置为 False 来忽略
  12.    """
  13.    # 存储返回值
  14.    ret = 0
  15.    if maxwidth:        if self.width &gt; maxwidth:
  16.            wpercent = (maxwidth / self.width)
  17.            hsize = int((self.height * wpercent))
  18.            fname = self.image.filename            # Image.LANCZOS 是重采样滤波器,用于抗锯齿
  19.            self.image = self.image.resize((maxwidth, hsize), Image.LANCZOS)
  20.            self.image.filename = fname
  21.            self.width, self.height = self.image.size
  22.            self.total_pixels = self.width * self.height
  23.            ret += 1
  24.    if maxheight:        if self.height &gt; maxheight:
  25.            hpercent = (maxheight / float(self.height))
  26.            wsize = int((float(self.width) * float(hpercent)))
  27.            fname = self.image.filename
  28.            self.image = self.image.resize((wsize, maxheight), Image.LANCZOS)
  29.            self.image.filename = fname
  30.            self.width, self.height = self.image.size
  31.            self.total_pixels = self.width * self.height
  32.            ret += 2
  33.    return ret

注:

Image.resize(size, resample=0)

size – 包含宽高像素数的元祖 (width, height) resample – 可选的重采样滤波器

返回 Image 对象

解析方法


   
  1. def parse(self):
  2.    # 如果已有结果,返回本对象
  3.    if self.result is not None:
  4.        return self    # 获得图片所有像素数据
  5.    pixels = self.image.load()

接着,遍历每个像素,为每个像素创建对应的 Skin 对象,代码见下

其中 self._classify_skin() 这个方法是检测像素颜色是否为肤色


   
  1.    for y in range(self.height):        for x in range(self.width):            # 得到像素的 RGB 三个通道的值
  2.            # [x, y] 是 [(x,y)] 的简便写法
  3.            r = pixels[x, y][0]   # red
  4.            g = pixels[x, y][1]   # green
  5.            b = pixels[x, y][2]   # blue
  6.            # 判断当前像素是否为肤色像素
  7.            isSkin = True if self._classify_skin(r, g, b) else False
  8.            # 给每个像素分配唯一 id 值(1, 2, 3...height*width)
  9.            # 注意 x, y 的值从零开始
  10.            _id = x + y * self.width + 1
  11.            # 为每个像素创建一个对应的 Skin 对象,并添加到 self.skin_map 中
  12.            self.skin_map.append(self.Skin(_id, isSkin, None, x, y))

若当前像素并不是肤色,那么跳过本次循环,继续遍历


   
  1.  # 若当前像素不为肤色像素,跳过此次循环      
  2.  if not isSkin:
  3.                 continue

若当前像素是肤色像素,那么就需要处理了,先遍历其相邻像素

一定要注意相邻像素的索引值,因为像素的 id 值是从 1 开始编起的,而索引是从 0 编起的。变量 _id 是存有当前像素的 id 值, 所以当前像素在 self.skin_map 中的索引值为 _id - 1,以此类推,那么其左方的相邻像素在 self.skin_map 中的索引值为 _id - 1 - 1 ,左上方为 _id - 1 - self.width - 1,上方为 _id - 1 - self.width ,右上方为 _id - 1 - self.width + 1


   
  1.  # 设左上角为原点,相邻像素为符号 *,当前像素为符号 ^,那么相互位置关系通常如下图 # *** # *^ # 存有相邻像素索引的列表,存放顺序为由大到小,顺序改变有影响 # 注意 _id 是从 1 开始的,对应的索引则是 _id-1
  2.            check_indexes = [_id - 2, # 当前像素左方的像素
  3.                             _id - self.width - 2,  # 当前像素左上方的像素
  4.                             _id - self.width - 1,  # 当前像素的上方的像素
  5.                             _id - self.width]  # 当前像素右上方的像素

说起来复杂,其实看上面代码并不复杂,说这么多是怕同学搞混,你要是觉得有点绕的话,你也可以把 id 值从 0 编起


   
  1.            # 用来记录相邻像素中肤色像素所在的区域号,初始化为 -1
  2.            region = -1
  3.            # 遍历每一个相邻像素的索引
  4.            for index in check_indexes:                # 尝试索引相邻像素的 Skin 对象,没有则跳出循环
  5.                try:
  6.                    self.skin_map[index]                except IndexError:                    break
  7.                # 相邻像素若为肤色像素:
  8.                if self.skin_map[index].skin:                    # 若相邻像素与当前像素的 region 均为有效值,且二者不同,且尚未添加相同的合并任务
  9.                    if (self.skin_map[index].region != None and
  10.                            region != None and region != -1 and
  11.                            self.skin_map[index].region != region and
  12.                            self.last_from != region and
  13.                            self.last_to != self.skin_map[index].region) :                        # 那么这添加这两个区域的合并任务
  14.                        self._add_merge(region, self.skin_map[index].region)                    # 记录此相邻像素所在的区域号
  15.                    region = self.skin_map[index].region

self._add_merge() 这个方法接收两个区域号,它将会把两个区域号添加到 self.merge_regions 中的元素中,self.merge_regions 的每一个元素都是一个列表,这些列表中存放了 1 到多个的区域号,区域号代表的区域是连通的,需要合并

检测的图像里,有些前几行的像素的相邻像素并没有 4 个,所以需要用 try “试错”

然后相邻像素的若是肤色像素,如果两个像素的皮肤区域号都为有效值且不同,因为两个区域中的像素相邻,那么其实这两个区域是连通的,说明需要合并这两个区域。记录下此相邻肤色像素的区域号,之后便可以将当前像素归到这个皮肤区域里了。

遍历完所有相邻像素后,分两种情况处理

所有相邻像素都不是肤色像素:

发现了新的皮肤区域

存在区域号为有效值的相邻肤色像素:

region 的中存储的值有用了,把当前像素归到这个相邻像素所在的区域


   
  1.            # 遍历完所有相邻像素后,若 region 仍等于 -1,说明所有相邻像素都不是肤色像素
  2.            if region == -1:                # 更改属性为新的区域号,注意元祖是不可变类型,不能直接更改属性
  3.                _skin = self.skin_map[_id - 1]._replace(region=len(self.detected_regions))
  4.                self.skin_map[_id - 1] = _skin                # 将此肤色像素所在区域创建为新区域
  5.                self.detected_regions.append([self.skin_map[_id - 1]])            # region 不等于 -1 的同时不等于 None,说明有区域号为有效值的相邻肤色像素
  6.            elif region != None:                # 将此像素的区域号更改为与相邻像素相同
  7.                _skin = self.skin_map[_id - 1]._replace(region=region)
  8.                self.skin_map[_id - 1] = _skin                # 向这个区域的像素列表中添加此像素
  9.                self.detected_regions[region].append(self.skin_map[_id - 1])
  • somenamedtuple._replace(kwargs) 返回一个替换指定字段的值为参数的 namedtuple 实例

遍历完所有像素之后,图片的皮肤区域划分初步完成了,只是在变量 self.merge_regions 中还有一些连通的皮肤区域号,它们需要合并,合并之后就可以进行色情图片判定了


   
  1.    # 完成所有区域合并任务,合并整理后的区域存储到 self.skin_regions
  2.    self._merge(self.detected_regions, self.merge_regions)    # 分析皮肤区域,得到判定结果
  3.    self._analyse_regions()
  4.        return self

方法 self._merge() 便是用来合并这些连通的皮肤区域的

方法 self._analyse_regions(),运用之前在程序原理一节定义的非色情图像判定规则,从而得到判定结果

现在编写我们还没写过的调用过的 Nude 类的方法

首先是 self._classify_skin() 方法,这个方法是检测像素颜色是否为肤色,之前在程序原理一节已经把肤色判定该公式列举了出来,现在是用的时候了


   
  1. # 基于像素的肤色检测
  2. def _classify_skin(self, r, g, b):
  3.    # 根据RGB值判定
  4.    rgb_classifier = r &gt; 95 and \
  5.        g &gt; 40 and g &lt; 100 and \
  6.        b &gt; 20 and \
  7.        max([r, g, b]) - min([r, g, b]) &gt; 15 and \
  8.        abs(r - g) &gt; 15 and \
  9.        r &gt; g and \
  10.        r &gt; b    # 根据处理后的 RGB 值判定
  11.    nr, ng, nb = self._to_normalized(r, g, b)
  12.    norm_rgb_classifier = nr / ng &gt; 1.185 and \
  13.        float(r * b) / ((r + g + b) ** 2) &gt; 0.107 and \
  14.        float(r * g) / ((r + g + b) ** 2) &gt; 0.112
  15.    # HSV 颜色模式下的判定
  16.    h, s, v = self._to_hsv(r, g, b)
  17.    hsv_classifier = h &gt; 0 and \
  18.        h &lt; 35 and \
  19.        s &gt; 0.23 and \
  20.        s &lt; 0.68
  21.    # YCbCr 颜色模式下的判定
  22.    y, cb, cr = self._to_ycbcr(r, g,  b)
  23.    ycbcr_classifier = 97.5 &lt;= cb &lt;= 142.5 and 134 &lt;= cr &lt;= 176
  24.    # 效果不是很好,还需改公式
  25.    # return rgb_classifier or norm_rgb_classifier or hsv_classifier or ycbcr_classifier
  26.    return ycbcr_classifier

颜色模式的转换并不是本实验的重点,转换公式可以在网上找到,这里我们直接拿来用就行


   
  1. def _to_normalized(self, r, g, b):
  2.    if r == 0:
  3.        r = 0.0001
  4.    if g == 0:
  5.        g = 0.0001
  6.    if b == 0:
  7.        b = 0.0001
  8.    _sum = float(r + g + b)    return [r / _sum, g / _sum, b / _sum]def _to_ycbcr(self, r, g, b):
  9.    # 公式来源:
  10.    # http://stackoverflow.com/questions/19459831/rgb-to-ycbcr-conversion-problems
  11.    y = .299*r + .587*g + .114*b
  12.    cb = 128 - 0.168736*r - 0.331364*g + 0.5*b
  13.    cr = 128 + 0.5*r - 0.418688*g - 0.081312*b    return y, cb, crdef _to_hsv(self, r, g, b):
  14.    h = 0
  15.    _sum = float(r + g + b)
  16.    _max = float(max([r, g, b]))
  17.    _min = float(min([r, g, b]))
  18.    diff = float(_max - _min)    if _sum == 0:
  19.        _sum = 0.0001
  20.    if _max == r:        if diff == 0:
  21.            h = sys.maxsize        else:
  22.            h = (g - b) / diff    elif _max == g:
  23.        h = 2 + ((g - r) / diff)    else:
  24.        h = 4 + ((r - g) / diff)
  25.    h *= 60
  26.    if h &lt; 0:
  27.        h += 360
  28.    return [h, 1.0 - (3.0 * (_min / _sum)), (1.0 / 3.0) * _max]

self._add_merge() 方法主要是对 self.merge_regions 操作,而self.merge_regions 的元素都是包含一些 int 对象(区域号)的列表,列表中的区域号代表的区域都是待合并的区域

self._add_merge() 方法接收两个区域号,将之添加到 self.merge_regions 中

这两个区域号以怎样的形式添加,要分3种情况处理,

  1. 传入的两个区域号都存在于 self.merge_regions 中

  2. 传入的两个区域号有一个区域号存在于 self.merge_regions 中

  3. 传入的两个区域号都不存在于 self.merge_regions 中

具体的处理方法,见代码


   
  1. def _add_merge(self, _from, _to):
  2.    # 两个区域号赋值给类属性
  3.    self.last_from = _from
  4.    self.last_to = _to    # 记录 self.merge_regions 的某个索引值,初始化为 -1
  5.    from_index = -1
  6.    # 记录 self.merge_regions 的某个索引值,初始化为 -1
  7.    to_index = -1
  8.    # 遍历每个 self.merge_regions 的元素
  9.    for index, region in enumerate(self.merge_regions):        # 遍历元素中的每个区域号
  10.        for r_index in region:            if r_index == _from:
  11.                from_index = index            if r_index == _to:
  12.                to_index = index    # 若两个区域号都存在于 self.merge_regions 中
  13.    if from_index != -1 and to_index != -1:        # 如果这两个区域号分别存在于两个列表中
  14.        # 那么合并这两个列表
  15.        if from_index != to_index:
  16.            self.merge_regions[from_index].extend(self.merge_regions[to_index])            del(self.merge_regions[to_index])        return
  17.    # 若两个区域号都不存在于 self.merge_regions 中
  18.    if from_index == -1 and to_index == -1:        # 创建新的区域号列表
  19.        self.merge_regions.append([_from, _to])        return
  20.    # 若两个区域号中有一个存在于 self.merge_regions 中
  21.    if from_index != -1 and to_index == -1:        # 将不存在于 self.merge_regions 中的那个区域号
  22.        # 添加到另一个区域号所在的列表
  23.        self.merge_regions[from_index].append(_to)        return
  24.    # 若两个待合并的区域号中有一个存在于 self.merge_regions 中
  25.    if from_index == -1 and to_index != -1:        # 将不存在于 self.merge_regions 中的那个区域号
  26.        # 添加到另一个区域号所在的列表
  27.        self.merge_regions[to_index].append(_from)        return

在序列中循环时,索引位置和对应值可以使用 enumerate() 函数同时得到,在上面的代码中,索引位置即为 index ,对应值即为region

self._merge() 方法则是将 self.merge_regions 中的元素中的区域号所代表的区域合并,得到新的皮肤区域列表


   
  1. def _merge(self, detected_regions, merge_regions):
  2.    # 新建列表 new_detected_regions
  3.    # 其元素将是包含一些代表像素的 Skin 对象的列表
  4.    # new_detected_regions 的元素即代表皮肤区域,元素索引为区域号
  5.    new_detected_regions = []    # 将 merge_regions 中的元素中的区域号代表的所有区域合并
  6.    for index, region in enumerate(merge_regions):        try:
  7.            new_detected_regions[index]        except IndexError:
  8.            new_detected_regions.append([])        for r_index in region:
  9.            new_detected_regions[index].extend(detected_regions[r_index])
  10.            detected_regions[r_index] = []    # 添加剩下的其余皮肤区域到 new_detected_regions
  11.    for region in detected_regions:        if len(region) &gt; 0:
  12.            new_detected_regions.append(region)    # 清理 new_detected_regions
  13.    self._clear_regions(new_detected_regions)        # 添加剩下的其余皮肤区域到 new_detected_regions
  14.        for region in detected_regions:            if len(region) &gt; 0:
  15.                new_detected_regions.append(region)        # 清理 new_detected_regions
  16.        self._clear_regions(new_detected_regions)

self._clear_regions() 方法只将像素数大于指定数量的皮肤区域保留到 self.skin_regions


   
  1. # 皮肤区域清理函数# 只保存像素数大于指定数量的皮肤区域def _clear_regions(self, detected_regions):
  2.    for region in detected_regions:        if len(region) &gt; 30:
  3.            self.skin_regions.append(region)

self._analyse_regions() 是很简单的,它的工作只是进行一系列判断,得出图片是否色情的结论


   
  1. # 分析区域
  2. def _analyse_regions(self):
  3.    # 如果皮肤区域小于 3 个,不是色情
  4.    if len(self.skin_regions) &lt; 3:
  5.        self.message = "Less than 3 skin regions ({_skin_regions_size})".format(
  6.            _skin_regions_size=len(self.skin_regions))
  7.        self.result = False
  8.        return self.result    # 为皮肤区域排序
  9.    self.skin_regions = sorted(self.skin_regions, key=lambda s: len(s),
  10.                               reverse=True)    # 计算皮肤总像素数
  11.    total_skin = float(sum([len(skin_region) for skin_region in self.skin_regions]))    # 如果皮肤区域与整个图像的比值小于 15%,那么不是色情图片
  12.    if total_skin / self.total_pixels * 100 &lt; 15:
  13.        self.message = "Total skin percentage lower than 15 ({:.2f})".format(total_skin / self.total_pixels * 100)
  14.        self.result = False
  15.        return self.result    # 如果最大皮肤区域小于总皮肤面积的 45%,不是色情图片
  16.    if len(self.skin_regions[0]) / total_skin * 100 &lt; 45:
  17.        self.message = "The biggest region contains less than 45 ({:.2f})".format(len(self.skin_regions[0]) / total_skin * 100)
  18.        self.result = False
  19.        return self.result    # 皮肤区域数量超过 60个,不是色情图片
  20.    if len(self.skin_regions) &gt; 60:
  21.        self.message = "More than 60 skin regions ({})".format(len(self.skin_regions))
  22.        self.result = False
  23.        return self.result    # 其它情况为色情图片
  24.    self.message = "Nude!!"
  25.    self.result = True
  26.    return self.result

然后可以组织下分析得出的信息


   
  1. def inspect(self):
  2.    _image = '{} {} {}×{}'.format(self.image.filename, self.image.format, self.width, self.height)    return "{_image}: result={_result} message='{_message}'".format(_image=_image, _result=self.result, _message=self.message)

Nude 类如果就这样完成了,最后运行脚本时只能得到一些真或假的结果,我们需要更直观的感受程序的分析效果,我们可以生成一张原图的副本,不过这个副本图片中只有黑白色,白色代表皮肤区域,那么这样我们能直观感受到程序分析的效果了

前面的代码中我们有获得图像的像素的 RGB 值的操作,设置像素的 RGB 值也就是其逆操作,还是很简单的,不过注意设置像素的 RGB 值时不能在原图上操作


   
  1.   # 将在源文件目录生成图片文件,将皮肤区域可视化
  2. def showSkinRegions(self):
  3.    # 未得出结果时方法返回
  4.    if self.result is None:
  5.        return
  6.    # 皮肤像素的 ID 的集合
  7.    skinIdSet = set()    # 将原图做一份拷贝
  8.    simage = self.image    # 加载数据
  9.    simageData = simage.load()    # 将皮肤像素的 id 存入 skinIdSet
  10.    for sr in self.skin_regions:        for pixel in sr:
  11.            skinIdSet.add(pixel.id)    # 将图像中的皮肤像素设为白色,其余设为黑色
  12.    for pixel in self.skin_map:        if pixel.id not in skinIdSet:
  13.            simageData[pixel.x, pixel.y] = 0, 0, 0
  14.        else:
  15.            simageData[pixel.x, pixel.y] = 255, 255, 255
  16.    # 源文件绝对路径
  17.    filePath = os.path.abspath(self.image.filename)    # 源文件所在目录
  18.    fileDirectory = os.path.dirname(filePath) + '/'
  19.    # 源文件的完整文件名
  20.    fileFullName = os.path.basename(filePath)    # 分离源文件的完整文件名得到文件名和扩展名
  21.    fileName, fileExtName = os.path.splitext(fileFullName)    # 保存图片
  22.    simage.save('{}{}_{}{}'.format(fileDirectory, fileName,'Nude' if self.result else 'Normal', fileExtName))

变量 skinIdSet 使用集合而不是列表是有性能上的考量的,Python 中的集合是哈希表实现的,查询效率很高

最后支持一下命令行参数就大功告成啦!我们使用 argparse 这个模块来实现命令行的支持。argparse 模块使得编写用户友好的命令行接口非常容易。程序只需定义好它要求的参数,然后 argparse 将负责如何从 sys.argv 中解析出这些参数。argparse 模块还会自动生成帮助和使用信息并且当用户赋给程序非法的参数时产生错误信息

具体使用方法请查看argparse的 官方文档,这里就不多说了


   
  1. if __name__ == "__main__":    
  2.   import argparse
  3.    parser = argparse.ArgumentParser(description='Detect nudity in images.')
  4.    parser.add_argument('files', metavar='image', nargs='+',
  5.                        help='Images you wish to test')
  6.    parser.add_argument('-r', '--resize', action='store_true',
  7.                        help='Reduce image size to increase speed of scanning')
  8.    parser.add_argument('-v', '--visualization', action='store_true',
  9.                        help='Generating areas of skin image')
  10.    args = parser.parse_args()    for fname in args.files:        if os.path.isfile(fname):
  11.            n = Nude(fname)            if args.resize:
  12.                n.resize(maxheight=800, maxwidth=600)
  13.            n.parse()            if args.visualization:
  14.                n.showSkinRegions()
  15.            print(n.result, n.inspect())        else:
  16.            print(fname, "is not a file")

4、运行代码:(准备两张图片,一张小黄图,一张比基尼)

python test.py -v a1.jpg a2.jpg

等待结果

640?wx_fmt=png

640?wx_fmt=png

5、over! python鉴黄工作完成

640?wx_fmt=png

所有代码:

https://www.bytelang.com/o/s/c/avlzMdzTSpE=

详情点击阅读原文查看

听说有气质的人都会关注这个公众号!

640?wx_fmt=jpeg

文章来源: blog.csdn.net,作者:敲代码的灰太狼,版权归原作者所有,如需转载,请联系作者。

原文链接:blog.csdn.net/tongtongjing1765/article/details/100581758

(完)