Python简单通过图片的颜色判断图片是否是情色图片的代码

"""
作者：皮蛋编程(https://www.pidancode.com)
创建日期：2022/3/19
修改日期：2022/3/19
功能描述：Python简单通过图片的颜色判断图片是否是情色图片的代码
"""

from PIL import Image
pic = 'test.jpg'
img = Image.open(pic).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.'%(pic, 'is' if cnt > w * h * 0.3 else 'is not'))