search

CharlesShang / TFFRCNN

FastER RCNN built on tensorflow
MIT License
874 stars 418 forks source link

ValueError: operands could not be broadcast together #69

Open hadign20 opened 7 years ago

hadign20 commented 7 years ago

I am trying to run faster rcnn on a custom dataset. After 5000 iterations suddenly this error shows up:

Traceback (most recent call last):
  File "./tools/train_net.py", line 96, in <module>
    max_iters=args.max_iters)
  File "/home/hadi/Software/tensorflow/TEST_FRCN_ROOT/tools/../lib/fast_rcnn/train.py", line 267, in train_net
    sw.train_model(sess, max_iters)
  File "/home/hadi/Software/tensorflow/TEST_FRCN_ROOT/tools/../lib/fast_rcnn/train.py", line 198, in train_model
    self.snapshot(sess, iter)
  File "/home/hadi/Software/tensorflow/TEST_FRCN_ROOT/tools/../lib/fast_rcnn/train.py", line 61, in snapshot
    sess.run(net.bbox_weights_assign, feed_dict={net.bbox_weights: orig_0 * np.tile(self.bbox_stds, (weights_shape[0], 1))})
ValueError: operands could not be broadcast together with shapes (4096,84) (4096,40)

My images are of size 1024*1024.

DonglinWang commented 6 years ago

Did U resolve this problem? I get the same one.