Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow training to succeed with negative images #102

Open
wants to merge 2 commits into
base: master
Choose a base branch
from

Conversation

nwestlake
Copy link

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.

Nick Westlake added 2 commits January 7, 2016 22:47
Added an assert statement to ensure that there is at least one label and roi
for the minibatch. If this is not the case, the training starts however results
in a floating point error which doesn't give the clear cause. This situation
can result from having images which are entirely negatives but not having
TRAIN.BG_THRESH_LO set to zero.
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.
@MinaRe
Copy link

MinaRe commented Apr 20, 2016

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

@panfengli
Copy link

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
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

4 participants