fast-rcnn icon indicating copy to clipboard operation
fast-rcnn copied to clipboard

Allow training to succeed with negative images

Open nwestlake opened this issue 9 years ago • 3 comments

Allows training to succeed when there are images present without any ground truth boxes. Previously, this failed due to merging empty matrices, argmax over empty matrices or the lack of an objective.

This supercedes 75b3fd (Assert at least one label for training) which would pick up the lack of an objective.

nwestlake avatar Jan 08 '16 02:01 nwestlake

Just put "sudo" before your comment, and if the problem still ,you can change number of batch.

MinaRe avatar Apr 20 '16 15:04 MinaRe

When output the model result, it will get this error:

Traceback (most recent call last):
  File "./tools/train_net.py", line 92, in <module>
    max_iters=args.max_iters)
  File "/home/hust/fast-rcnn/tools/../lib/fast_rcnn/train.py", line 123, in train_net
    sw.train_model(max_iters)
  File "/home/hust/fast-rcnn/tools/../lib/fast_rcnn/train.py", line 98, in train_model
    self.snapshot()
  File "/home/hust/fast-rcnn/tools/../lib/fast_rcnn/train.py", line 56, in snapshot
    orig_0 = net.params['bbox_pred'][0].data.copy()
KeyError: 'bbox_pred'

panfengli avatar Jun 06 '16 15:06 panfengli

Oh, I find it is another error, since I have renamed bbox_pred to bbox_pred_inria in prototxt

panfengli avatar Jun 06 '16 15:06 panfengli