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

因为要使用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')

你可能感兴趣的:(Deep,Learning)