python - 保存为灰色的增强图像(imgaug、imageio、opencv)
问题描述
我正在开始我的第一个计算机视觉项目,并想尝试图像增强来改进我的数据集。我用 imgaug 试了一下,一切正常,只要我只显示它们。但是,一旦我将它们保存到文件中,它们就会变成灰色/蓝色。
这是我正在使用的代码:
import cv2
import imgaug.augmenters as iaa
import imageio as iio
import numpy as np
path = ('C:\\Users\\kathr\\Desktop\\JPG\\cards.jpg')
img = cv2.imread(path)
augmentation= iaa.Sequential([
iaa.Crop(px=(0,2)),
iaa.Flipud(0.5),
iaa.Fliplr(0.5),
iaa.Sometimes(
0.5,
iaa.GaussianBlur(sigma=(0,0.5)),
)] ,
random_order = True)
image = cv2.imread(path)
cv2.imshow('orign', img)
cv2.waitKey(100)
images = np.array([image for _ in range(32)], dtype=np.uint8)
images_aug = augmentation.augment_images(images)
for i in range(32):
iio.imwrite('C:\\Users\\kathr\\Desktop\\Aug\\' + str(i) + 'aug.jpg', images_aug[i], format= '.jpg')
解决方案
Opencv 按 B、G、R 顺序读取图像。在保存之前将它们转换为 R、G、B 将使用 imageio 解决您的问题。查看这篇文章了解更多信息。
for i in range(32):
images_aug[i] = cv2.cvtColor(images_aug[i], cv2.COLOR_BGR2RGB)
iio.imwrite('C:\\Users\\kathr\\Desktop\\Aug\\' + str(i) + 'aug.jpg', images_aug[i], format= '.jpg')
由于您已经在使用 opencv,您也可以使用 opencv 直接保存它们,避免转换。
for i in range(32):
cv2.imwrite('C:\\Users\\kathr\\Desktop\\Aug\\' + str(i) + 'aug.jpg', images_aug[i])
推荐阅读
- python-3.x - argparse 位置参数 '*'
- html - 如何将内部的 div 边框触摸到外部的 div 底部?
- node.js - 让 https express 节点应用程序在共享主机上工作
- firebase - 为什么删除所有文档和集合时,firestore 不删除存储的字节和对象?
- ssl - Branch.io - 设置自定义根域
- git - 一个文件可以由git中的多个blob组成吗?
- haskell - XMonad 按键功能如何减少?
- laravel - 带有确认页面的 Laravel 联系表
- vue.js - 如果我导入 printjs,vuejs 组件不会呈现
- android - 如何仅在纵向模式下设置 libgdx livewallpaper 屏幕方向?