object_detection_demo
object_detection_demo copied to clipboard
[Help] Cannot train using faster_rcnn_inception_v2 or rfcn_resnet101
I can train with ssd_mobilenet_v2 but cannot train with faster_rcnn_inception_v2 or rfcn_resnet101 I got the below error. Can any one help me! Thank you so much!
Traceback (most recent call last):
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 1365, in _do_call
return fn(*args)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 1350, in _run_fn
target_list, run_metadata)
File "/tensorflow-1.15.2/python3.7/tensorflow_core/python/client/session.py", line 1443, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: 2 root error(s) found.
(0) Invalid argument: Cannot add tensor to the batch: number of elements does not match. Shapes are: [tensor]: [289,1024,3], [batch]: [551,1024,3]
[[{{node IteratorGetNext}}]]
[[Loss/RPNLoss/Loss/huber_loss/assert_broadcastable/is_valid_shape/has_valid_nonscalar_shape/has_invalid_dims/num_invalid_dims/_6571]]
(1) Invalid argument: Cannot add tensor to the batch: number of elements does not match. Shapes are: [tensor]: [289,1024,3], [batch]: [551,1024,3]
[[{{node IteratorGetNext}}]]