用 Python 鑒別色色的圖片

YancyYe 發(fā)布于2019-07-30 16:33 / 2599人閱讀

摘要：代碼識(shí)別少量圖片先導(dǎo)入這個(gè)庫我也很奇怪為什么的是再寫出代碼代碼運(yùn)行的結(jié)果居然是讓我們來看一下圖片。說明識(shí)別是不是色色的圖片的有依據(jù)的，根據(jù)給出的信息來看，是皮膚暴露的百分比來判斷的，暴露的比例大于就判斷為色色的圖片。

0 前言

實(shí)話實(shí)說啊，這個(gè)標(biāo)題起得就有點(diǎn)標(biāo)題黨，識(shí)別是識(shí)別，準(zhǔn)確率就有點(diǎn)玄學(xué)了。

1 環(huán)境說明

Win10 系統(tǒng)下 Python3，編譯器是 Pycharm，需要安裝 nonude 這個(gè)庫。

Pycharm 安裝第三方庫的方法。

2 代碼 2.1 識(shí)別少量圖片

先導(dǎo)入 nonude 這個(gè)庫（我也很奇怪為什么 import 的是 nude）

import nude

再寫出代碼

print(nude.is_nude("godfather.jpg"))
print(nude.is_nude("leisheng.jpg"))
print(nude.is_nude("qiaoba.png"))

代碼運(yùn)行的結(jié)果居然是 False、True、True
讓我們來看一下圖片。

對(duì)了，我這里是直接把圖片放在了項(xiàng)目的路徑里的

如果不喜歡這樣，可以換一個(gè)寫法

import nude
print(nude.is_nude("E:/Images/OOXX/00mitai.jpg"))
print(nude.is_nude("E:ImagesOOXX/00zkted.jpg"))
print(nude.is_nude("E:/ImagesOOXX/011idk8.jpg"))

代碼運(yùn)行的結(jié)果是 False、True、False

說一下，路徑里面的斜杠 / 可以是反斜杠，如果后面有數(shù)字的話，請(qǐng)使用斜杠 /，不要用反斜杠。

2.2 識(shí)別文件夾中的圖片

先導(dǎo)入要用的庫

import glob
import itertools
from nude import Nude

這里用 glob 返回匹配指定模式的文件名

images_format = ["jpg", "png", "gif"]  # 圖片格式
images_jpg = glob.glob("E:/Images/OOXX/*.jpg")  # 返回匹配指定模式的文件名
images_png = glob.glob("E:/Images/OOXX/*.png")
images_gif = glob.glob("E:/Images/OOXX/*.gif")

itertools.chain 把迭代對(duì)象串聯(lián)起來，形成一個(gè)更大的迭代器，說白了就是把 images_jpg、images_png、images_gif 弄在一起，變成了images_list。

然后進(jìn)行循環(huán)，在 images_list 里面一個(gè)一個(gè)對(duì)圖片進(jìn)行識(shí)別。

images_list = itertools.chain(images_jpg, images_png, images_gif)

for i in images_list:
    print(i)  # 輸出照片的路徑
    n = Nude(i)  # 對(duì)圖片進(jìn)行識(shí)別
    n.parse()
    print(n.result)  # 輸出結(jié)果
    print(n.message)  # 輸出判斷信息
    print(n.inspect())  # 輸出更加詳細(xì)的判斷信息

運(yùn)行結(jié)果如下圖

我們來看一下下圖片，準(zhǔn)確率嘛，有點(diǎn)玄學(xué)。

3 說明

識(shí)別是不是色色的圖片的有依據(jù)的，根據(jù)給出的信息來看，是皮膚暴露的百分比來判斷的，暴露的比例大于 15% 就判斷為色色的圖片。這里只是很粗淺的說明，還有一些我沒有看懂。

nonude 的 GitHub 鏈接：https://github.com/gearsystems/nonude/

nonude 的最近一次更新是兩年前了，感覺有點(diǎn)可惜，希望作者能繼續(xù)維護(hù)。

文章首發(fā)于本人公眾號(hào)，歡迎關(guān)注~