将一个文件夹内的灰度图片变为彩色图片python代码实现
时间: 2023-05-18 20:07:09 浏览: 91
以下是将一个文件夹内的灰度图片变为彩色图片的 Python 代码实现:
```python
import os
import cv2
# 定义函数将灰度图转换为彩色图
def gray2color(gray_img):
color_img = cv2.cvtColor(gray_img, cv2.COLOR_GRAY2BGR)
return color_img
# 定义函数将文件夹内的所有灰度图转换为彩色图
def convert_gray2color_folder(input_folder, output_folder):
# 确保输出文件夹存在
if not os.path.exists(output_folder):
os.makedirs(output_folder)
# 遍历输入文件夹内的所有文件
for filename in os.listdir(input_folder):
# 仅处理灰度图
if filename.endswith('.jpg') or filename.endswith('.png'):
gray_img = cv2.imread(os.path.join(input_folder, filename), cv2.IMREAD_GRAYSCALE)
color_img = gray2color(gray_img)
cv2.imwrite(os.path.join(output_folder, filename), color_img)
# 调用函数将灰度图转换为彩色图
convert_gray2color_folder('input_folder', 'output_folder')
```
请注意,此代码需要 OpenCV 库的支持。
阅读全文