tensorflow和Keras 转换RGB到BGR的实现代码

最新推荐文章于 2024-06-21 21:17:27 发布

Will_Ye

最新推荐文章于 2024-06-21 21:17:27 发布

阅读量3.5k

点赞数

分类专栏： Deep Learning 文章标签： RGB2BGR

本文链接：https://blog.csdn.net/will_ye/article/details/88062567

版权

Deep Learning 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

因为要使用opencv处理图像，就必须将图片的格式从RGB的格式转换成BGR的形式，这里列一下用得比较多TF和Keras版本代码。

tensorflow

import numpy as np
import tensorflow as tf

vgg_mean = [103.939, 116.779, 123.68]

self.tfx = tf.placeholder(tf.float32, [None, 224, 224, 3])
self.tfy = tf.placeholder(tf.float32, [None, 1])

# Convert RGB to BGR
red, green, blue = tf.split(axis=3, num_or_size_splits=3, value=self.tfx * 255.0)
bgr = tf.concat(axis=3, values=[
	blue - self.vgg_mean[0],
    green - self.vgg_mean[1],
    red - self.vgg_mean[2],
])

Keras

这里用到VGG16和手写数字识别mnist数据集

import cv2
from keras import datasets
from keras.applications.vgg16 import VGG16
from keras.datasets import mnist
import numpy as np


(X_train,y_train),(X_test,y_test) = mnist.load_data()
 
#转成VGG16需要的格式
#RGB ->> bgr格式
X_train = [cv2.cvtColor(cv2.resize(i,(ishape,ishape)), cv2.COLOR_GRAY2BGR) for i in X_train]
X_train = np.concatenate([arr[np.newaxis] for arr in X_train]).astype('float32')
 
X_test  = [cv2.cvtColor(cv2.resize(i,(ishape,ishape)), cv2.COLOR_GRAY2BGR) for i in X_test ]
X_test  = np.concatenate([arr[np.newaxis] for arr in X_test] ).astype('float32')