python鑒黃程式

標籤：os.path   目標   sdi   服務   not   for   pat   客戶   col   

最近有客戶向伺服器上傳了些垃圾圖片，和正常圖片混合在一起，大概有10W張的數量，在經曆了大概3個小時翻了2000多張的時候，手指抽了下，感覺很不舒服，硬著頭皮上，寫個程式鑒別下吧，但是怎麼搞呢，經過從網上翻閱資料和同事的協助，終於把這個命中率不高的程式弄好了：

代碼如下：

# -*- coding: utf8 -*-# Author:wxq#python 2.7import sysimport osimport os.path as pimport shutilimport _iofrom collections import namedtuplefrom PIL import Imagedef check_porn(file):    img = Image.open(file).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    if cnt > w * h * 0.1:        return True    else:        return Falseif __name__ == ‘__main__‘:    imgdir = sys.argv[1]    dstdir = sys.argv[2]    if imgdir == ‘‘ or dstdir == ‘‘:        raise Exception(‘參數不可為空, 用法: porn_check.py <圖片目錄> <目標目錄>‘)    if p.isdir(imgdir):        filelist = [p.join(imgdir, file) for file in os.listdir(imgdir)]    else:        raise Exception(‘參數錯誤, 圖片目錄不合法‘)        sys.exit(1)    if not p.isdir(dstdir):        raise Exception(‘參數錯誤, 目標目錄不合法‘)    for file in filelist:        if p.isfile(file):            is_porn = check_porn(file)            if is_porn:                print(file, ‘is porn‘)                shutil.move(file, dstdir)            else:                print(file, ‘is not porn‘)

正在空餘時間學習python，期待最佳化，爭取把鑒黃率提高至100%，杜絕一切危害公司利益和危害公用網路安全的事情發生。

python鑒黃程式