search

oarriaga / STN.keras

Implementation of spatial transformer networks (STNs) in keras 2 with tensorflow as backend.
MIT License
278 stars 75 forks source link

STN example not working #19

Open sumeetssaurav opened 3 years ago

sumeetssaurav commented 3 years ago

Traceback (most recent call last): File "train.py", line 16, in model = STN() File "/home/vision/STN/STN.py", line 29, in STN x = Conv2D(32, (3, 3), padding='same')(interpolated_image) File "/home/vision/.virtualenvs/dl4cv/lib/python3.5/site-packages/tensorflow_core/python/keras/engine/base_layer.py", line 812, in call self.name) File "/home/vision/.virtualenvs/dl4cv/lib/python3.5/site-packages/tensorflow_core/python/keras/engine/input_spec.py", line 155, in assert_input_compatibility ' input tensors. Inputs received: ' + str(inputs)) ValueError: Layer conv2d_2 expects 1 inputs, but it received 4 input tensors. Inputs received: [<tf.Tensor 'bilinear_interpolation/Placeholder:0' shape= dtype=float32>, <tf.Tensor 'bilinear_interpolation/Placeholder_1:0' shape=(30,) dtype=float32>, <tf.Tensor 'bilinear_interpolation/Placeholder_2:0' shape=(30,) dtype=float32>, <tf.Tensor 'bilinear_interpolation/Placeholder_3:0' shape=(1,) dtype=float32>]

JakobPCoder commented 2 years ago

Same problem :/ cant find the problem

zjhJOJO commented 2 years ago

same problem.