Могу ли я использовать `tf.layers.conv3d` с разными размерами изображения?Использовать None вместо целого числа исправления? - PullRequest
Новая
       10

Могу ли я использовать `tf.layers.conv3d` с разными размерами изображения?Использовать None вместо целого числа исправления?

0 голосов
/ 11 октября 2018

Следующий скрипт не работает: 

import tensorflow as tf
import numpy as np 
input = np.ones((1,1, 3,50,50))
t_ph = tf.placeholder(dtype=tf.float32, shape=(1,1,3,None,50), name='input')
# t_ph = tf.placeholder(dtype=tf.float32, shape=(1,1,3,50,50), name='input')
t_conv3d = tf.layers.conv3d(
    inputs=t_ph,
    filters=48,
    kernel_size=[3,3,3],
    data_format='channels_first',
    activation=tf.nn.relu
)

with tf.Session() as sess: 
    sess.run([tf.local_variables_initializer(), tf.global_variables_initializer()])

    should_fail = t_conv3d.eval(feed_dict={t_ph: input})
    print "should_fail ", should_fail.shape

Ошибка: TypeError: unsupported operand type(s) for *: 'int' and 'NoneType'

Возможно ли использовать conv3d с параметром 'None' вместо целого числа?

...