python用opencv图像变换_将python opencv mat图像转换为tensorflow图像d

使用imread加载OpenCV图像,然后将其转换为numpy数组。

要进入inception v3,需要使用Mult:0张量作为入口点,这需要一个具有布局的四维张量:[Batch index,Width,Height,Channel]

从cv::Mat来看,后三个非常好,第一个只需要0,因为您不想输入一批图像,而是一个图像。

代码如下:#Loading the file

img2 = cv2.imread(file)

#Format for the Mul:0 Tensor

img2= cv2.resize(img2,dsize=(299,299), interpolation = cv2.INTER_CUBIC)

#Numpy array

np_image_data = np.asarray(img2)

#maybe insert float convertion here - see edit remark!

np_final = np.expand_dims(np_image_data,axis=0)

#now feeding it into the session:

#[... initialization of session and loading of graph etc]

predictions = sess.run(softmax_tensor,

{'Mul:0': np_final})

#fin!

谨致问候

克里斯

编辑:我刚刚注意到,inception network需要将强度值规范化为浮动值[-0.5,0.5],因此在构建RGB图像之前,请使用以下代码对其进行转换:np_image_data=cv2.normalize(np_image_data.astype('float'), None, -0.5, .5, cv2.NORM_MINMAX)

你可能感兴趣的:(python用opencv图像变换_将python opencv mat图像转换为tensorflow图像d)