批量将彩色图像转化为灰度图

# -*- 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为指定路径。

转化前:

转化后:

注意!该脚本会修改原图像,因此如需要原图的话,最好留一份备份。

posted @ 2019-04-09 09:59  小小小小小码农  阅读(2447)  评论(0编辑  收藏  举报