使用Conv2d對圖像進行調整

Question

我正在使用TensorFlow處理與CNN相關的項目。 我輸入使用圖像（20個這樣的圖像）

for filename in glob.glob('input_data/*.jpg'): 
input_images.append(cv2.imread(filename,0)) 

image_size_input = len(input_images[0]) 

這些圖像是尺寸（250250），因爲灰度的。 但是對於conv2D，它需要一個4D輸入張量來饋送。我的輸入張量看起來像

x = tf.placeholder(tf.float32,shape=[None,image_size_output,image_size_output,1], name='x') 

所以我無法將上面的2D圖像轉換成給定的形狀（4D）。如何處理「無」字段。 我嘗試這樣做：

input_images_padded = [] 
for image in input_images: 
temp = np.zeros((1,image_size_output,image_size_output,1)) 
for i in range(image_size_input): 
    for j in range(image_size_input): 
     temp[0,i,j,0] = image[i,j] 
input_images_padded.append(temp) 

我得到了以下錯誤：

File "/opt/intel/intelpython3/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 975, in _run 
% (np_val.shape, subfeed_t.name, str(subfeed_t.get_shape()))) 

ValueError: Cannot feed value of shape (20, 1, 250, 250, 1) for Tensor 'x_11:0', which has shape '(?, 250, 250, 1)' 

這裏的整個代碼（僅供參考）：

import tensorflow as tf 
from PIL import Image 
import glob 
import cv2 
import os 
import numpy as np 
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' 

input_images = [] 
output_images = [] 

for filename in glob.glob('input_data/*.jpg'): 
    input_images.append(cv2.imread(filename,0)) 

for filename in glob.glob('output_data/*.jpg'): 
    output_images.append(cv2.imread(filename,0))  

image_size_input = len(input_images[0]) 
image_size_output = len(output_images[0]) 

''' 
now adding padding to the input images to convert from 125x125 to 250x2050 sized images 
''' 
input_images_padded = [] 
for image in input_images: 
    temp = np.zeros((1,image_size_output,image_size_output,1)) 
    for i in range(image_size_input): 
     for j in range(image_size_input): 
      temp[0,i,j,0] = image[i,j] 
    input_images_padded.append(temp) 

output_images_padded = [] 
for image in output_images: 
    temp = np.zeros((1,image_size_output,image_size_output,1)) 
    for i in range(image_size_input): 
     for j in range(image_size_input): 
      temp[0,i,j,0] = image[i,j] 
    output_images_padded.append(temp) 



sess = tf.Session() 
''' 
Creating tensor for the input 
''' 
x = tf.placeholder(tf.float32,shape= [None,image_size_output,image_size_output,1], name='x') 
''' 
Creating tensor for the output 
''' 
y = tf.placeholder(tf.float32,shape= [None,image_size_output,image_size_output,1], name='y') 


def create_weights(shape): 
    return tf.Variable(tf.truncated_normal(shape, stddev=0.05)) 

def create_biases(size): 
    return tf.Variable(tf.constant(0.05, shape=[size])) 

def create_convolutional_layer(input, bias_count, filter_height, filter_width, num_input_channels, num_out_channels, activation_function): 


    weights = create_weights(shape=[filter_height, filter_width, num_input_channels, num_out_channels]) 

    biases = create_biases(bias_count) 


    layer = tf.nn.conv2d(input=input, 
        filter=weights, 
       strides=[1, 1, 1, 1], 
       padding='SAME') 

    layer += biases 


layer = tf.nn.max_pool(value=layer, 
         ksize=[1, 2, 2, 1], 
         strides=[1, 1, 1, 1], 
         padding='SAME') 

if activation_function=="relu": 
    layer = tf.nn.relu(layer) 

return layer 


''' 
Conv. Layer 1: Patch extraction 
64 filters of size 1 x 9 x 9 
Activation function: ReLU 
Output: 64 feature maps 
Parameters to optimize: 
    1 x 9 x 9 x 64 = 5184 weights and 64 biases 
''' 
layer1 = create_convolutional_layer(input=x, 
           bias_count=64, 
           filter_height=9, 
           filter_width=9, 
           num_input_channels=1, 
           num_out_channels=64, 
           activation_function="relu") 

''' 
Conv. Layer 2: Non-linear mapping 
32 filters of size 64 x 1 x 1 
Activation function: ReLU 
Output: 32 feature maps 
Parameters to optimize: 64 x 1 x 1 x 32 = 2048 weights and 32 biases 
''' 

layer2 = create_convolutional_layer(input=layer1, 
           bias_count=32, 
           filter_height=1, 
           filter_width=1, 
           num_input_channels=64, 
           num_out_channels=32, 
           activation_function="relu") 

'''Conv. Layer 3: Reconstruction 
1 filter of size 32 x 5 x 5 
Activation function: Identity 
Output: HR image 
Parameters to optimize: 32 x 5 x 5 x 1 = 800 weights and 1 bias''' 
layer3 = create_convolutional_layer(input=layer2, 
           bias_count=1, 
           filter_height=5, 
           filter_width=5, 
           num_input_channels=32, 
           num_out_channels=1, 
           activation_function="identity") 

'''print(layer1.get_shape().as_list()) 
print(layer2.get_shape().as_list()) 
print(layer3.get_shape().as_list())''' 

''' 
    applying gradient descent algorithm 
''' 
#loss_function 
loss = tf.reduce_sum(tf.square(layer3-y)) 
#optimiser 
optimizer = tf.train.GradientDescentOptimizer(0.01) 
train = optimizer.minimize(loss) 


init = tf.global_variables_initializer() 
sess.run(init) 
for i in range(len(input_images)): 
    sess.run(train,{x: input_images_padded, y:output_images_padded}) 


curr_loss = sess.run([loss], {x: x_train, y: y_train}) 
print("loss: %s"%(curr_loss)) 

Answer 1

我認爲你的image_padded不正確。我沒有tf代碼寫作經驗（儘管已經閱讀了一些代碼）。但試試這個：

// imgs is your input-image-sequences 
// padded is to feed 
cnt = len(imgs) 
H,W = imgs[0].shape[:2] 
padded = np.zeros((cnt, H, W, 1)) 
for i in range(cnt): 
    padded[i, :,:,0] = img[i] 

Answer 2

一種選擇是忽略給予shape當您創建佔位符時，它會接受您在sess.run()

期間餵食的任何形狀的張量

從文檔：

shape: The shape of the tensor to be fed (optional). If the shape is not specified, you can feed a tensor of any shape.

或者，您可以指定20，這是你的批量大小。請注意張量中的第一維總是對應於batch_size