deandon / fpn-keras Goto Github PK

Caused by op 'rpn_bbox_loss/sub', defined at:
File "train_VOC_fpn.py", line 210, in
model_dir=MODEL_DIR)
File "/media/wjl/back/FPN-keras/model_FPN.py", line 1758, in init
self.keras_model = self.build(mode=mode, config=config)
File "/media/wjl/back/FPN-keras/model_FPN.py", line 1902, in build
[input_rpn_bbox, input_rpn_match, rpn_bbox])
File "/usr/local/lib/python3.5/dist-packages/keras/engine/topology.py", line 603, in call
output = self.call(inputs, **kwargs)
File "/usr/local/lib/python3.5/dist-packages/keras/layers/core.py", line 651, in call
return self.function(inputs, **arguments)
File "/media/wjl/back/FPN-keras/model_FPN.py", line 1901, in
rpn_bbox_loss = KL.Lambda(lambda x: rpn_bbox_loss_graph(config, *x), name="rpn_bbox_loss")(
File "/media/wjl/back/FPN-keras/model_FPN.py", line 1078, in rpn_bbox_loss_graph
diff = K.abs(target_bbox - rpn_bbox)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/ops/math_ops.py", line 894, in binary_op_wrapper
return func(x, y, name=name)
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/ops/gen_math_ops.py", line 4636, in _sub
"Sub", 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: [4,4] vs. [6,4]

Abandoned?

CLASS = ['__backgroud', 'OK','Deformation','Rupture','Sticky','Missing corner']