批量將彩色圖像轉化為灰度圖

# -*- coding:utf-8 -*-
import cv2
import os

input_dir = '/home/wmy/darknet/VOCdevkit3/VOC2007/JPEGImages'
out_dir = '/home/wmy/darknet/VOCdevkit3/VOC2007/JPEGImages'
a = os.listdir(input_dir)
for i in a:
    img = cv2.imread(input_dir+'/'+i)
    gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
    cv2.imencode('.jpg', gray)[1].tofile(out_dir+'/'+i)

在python3中，cv2.write()函數已不適用，因此可用cv2.imencode（）函數替代，tofile為指定路徑。

轉化前：

轉化后：

注意！該腳本會修改原圖像，因此如需要原圖的話，最好留一份備份。