maxritter / sdc-semantic-segmentation Goto Github PK

Hi @maxritter!

Great repo, informative README and clean code.

I'm trying to change the image shape from (160, 576) to (676, 844), but I get an error:
...

Caused by op 'Add', defined at:
File "main.py", line 187, in
run()
File "main.py", line 143, in run
layer_output = layers(layer3_out, layer4_out, layer7_out, num_classes)
File "main.py", line 51, in layers
skip_1 = tf.add(deconv_1, conv_layer4)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/ops/gen_math_ops.py", line 183, in add
"Add", x=x, y=y, name=name)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 2956, in create_op
op_def=op_def)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/framework/ops.py", line 1470, in init
self._traceback = self._graph._extract_stack() # pylint: disable=protected-access
InvalidArgumentError (see above for traceback): Incompatible shapes: [1,44,54,2] vs. [1,43,53,2]
[[Node: Add = Add[T=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:GPU:0"](conv2d_transpose/BiasAdd, conv2d_2/BiasAdd)]]
[[Node: Mean/_79 = _Recvclient _terminated=false, recv_device="/job:localhost/replica:0/task:0/device:CPU:0", send_device="/job:localhost/replica:0/task:0/device:GPU:0", send_device_incarnation=1, tensor_name="edge_1709_Mean", tensor_type=DT_FLOAT, _device="/job:localhost/replica:0/task:0/device:CPU:0"]]

This isn't really an issue, but what part of the code has to be changed to get this working for my shape?