object_detection_demo icon indicating copy to clipboard operation
object_detection_demo copied to clipboard

[Help] Cannot train using faster_rcnn_inception_v2 or rfcn_resnet101

Open KienPM opened this issue 3 years ago • 0 comments

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}}]]

KienPM avatar Apr 24 '21 14:04 KienPM