首页 ›电脑› 电脑学堂 › 查看内容

python用10行代码实现对黄色图片的检测功能

2015-8-10 17:30 1743 0

摘要: 本文实例讲述了python用10行代码实现对黄色图片的检测功能。分享给大家供大家参考。具体如下：原理：将图片转换为YCbCr模式，在图片中寻找图片色值像素，如果在皮肤色值内的像素面积超过整个画面的1/3，就认为是黄 ...

关键词： nbsp 图片像素 YCbCr 黄色代码 python 肤色 Image porn

本文实例讲述了python用10行代码实现对黄色图片的检测功能。分享给大家供大家参考。具体如下：

原理：将图片转换为YCbCr模式，在图片中寻找图片色值像素，如果在皮肤色值内的像素面积超过整个画面的1/3，就认为是黄色图片。

申明：简单场景还是够用了，稍微复杂一点就不准确了，例如：整幅画面是人的头像，皮肤色值的像素必然超过50%，被误认为黄色图片就太武断了。

需要安装python图片库PIL支持

porn_detect.py如下：

import sys,PIL.Image as Image
img = Image.open(sys.argv[1]).convert('YCbCr')
w, h = img.size
data = img.getdata()
cnt = 0
for i, ycbcr in enumerate(data):
  y, cb, cr = ycbcr
  if 86 <= cb <= 117 and 140 <= cr <= 168:
    cnt += 1
print '%s %s a porn image.'%(sys.argv[1], 'is' if cnt > w * h * 0.3 else 'is not')