Skip to content

Commit d659dbd

Browse files
authored
update documents and offset initialisation for the deformable CNN
update documents and offset initialisation for the deformable CNN
1 parent 7f1acd4 commit d659dbd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorlayer/layers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1935,7 +1935,7 @@ def __init__(
19351935
grid = tf.tile(grid, [1, 1, kernel_n, 1]) # grid --> (h, w, n, 2)
19361936
grid_offset = grid + initial_offsets # grid_offset --> (h, w, n, 2)
19371937

1938-
input_deform = efficient_tf_batch_map_offsets(self.inputs, offset, grid_offset)
1938+
input_deform = tf_batch_map_offsets(self.inputs, offset, grid_offset)
19391939

19401940
W = tf.get_variable(name='W_conv2d', shape=[1, 1, shape[0] * shape[1], shape[-2], shape[-1]],
19411941
initializer=W_init, **W_init_args)

0 commit comments

Comments
 (0)