Как построить spp-net в Tensorflow?

Question

Поскольку tf.max_pool dosen't поддерживает переменный размер, уровень spp не работает. Я передаю длину на график, получил ошибку.

length = tf.placeholder(tf.int32) 
shape=[-1, length,length,32] 
pool = tf.nn.max_pool(output, 
        ksize=[1, np.ceil(shape[1] * 1./l).astype(np.int32), np.ceil(shape[2] * 1./l).astype(np.int32), 1], 
        strides=[1, np.floor(shape[1] * 1./l + 1).astype(np.int32), np.floor(shape[2] * 1./l + 1), 1], 
          padding='SAME') 

ТипError: Ожидается, что int32 получил 1.0 типа 'float' вместо этого.

У вас есть идея, как с этим бороться?

Answer 1

В параметре strides измените третье значение на целое. т.е.

np.floor(shape[2] * 1./l + 1) в np.floor(shape[2] * 1./l + 1).astype(np.int32)

Полного stride варианта:

strides=[1, np.floor(shape[1] * 1./l + 1).astype(np.int32), 
      np.floor(shape[2] * 1./l + 1).astype(np.int32), 1], 

Надеется, что это помогает вам.