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

BUG? ground truth BoxList initialization error when using it for relation prediction? #102

Open
shizi626 opened this issue Aug 25, 2020 · 0 comments

Comments

@shizi626
Copy link

When using relation heads (lib\scene_parser\rcnn\modeling\relation_heads\relation_heads.py) to predict relations with ground truth bounding box in training phrase, I got a weird error in here

  File "/home/graph-rcnn.pytorch-master/lib/scene_parser/rcnn/structures/boxlist_ops.py", line 123, in cat_boxlist
    assert all(set(bbox.fields()) == fields for bbox in bboxes)
AssertionError

After some survey, I found the reason is that the fields in proposals and proposals_gt is different after Line 144 proposals_gt = self.box_post_processor((class_logits, box_regression), targets_cp, skip_nms=True) in relation_heads.py.

(Pdb) proposals_gt[0].fields()
['scores', 'logits', 'features', 'labels', 'regression_targets']
(Pdb) proposals[0].fields()
['objectness', 'labels', 'regression_targets', 'features', 'logits']

The proposals_gt is expected to be initialized the same as proposals in PostProcessor in lib\scene_parser\rcnn\modeling\roi_heads\box_head\inference.py, however, Line 110 add one field named scores rather than objectness. Is this a bug? Or do I misunderstand the usage?

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

No branches or pull requests

1 participant