[fix] Fix detected objects in extract_features_vmb (#159)

rmrao · apsdehal · commit 2df9160c6daa · 2019-08-22T14:47:29.000-07:00
* [fix] Correctly ignore the background class

* [fix] Output the correct number of objects
diff --git a/pythia/scripts/features/extract_features_vmb.py b/pythia/scripts/features/extract_features_vmb.py
@@ -162,7 +162,7 @@ def _process_feature_extraction(
             feat_list.append(feats[i][keep_boxes])
             bbox = output[0]["proposals"][i][keep_boxes].bbox / im_scales[i]
             # Predict the class label using the scores
-            objects = torch.argmax(scores[keep_boxes][start_index:], dim=1)
+            objects = torch.argmax(scores[keep_boxes][:, start_index:], dim=1)
 
             info_list.append(
                 {

Original file line number	Diff line number	Diff line change
`@@ -162,7 +162,7 @@ def _process_feature_extraction(`
`162`	`162`	`feat_list.append(feats[i][keep_boxes])`
`163`	`163`	`bbox = output[0]["proposals"][i][keep_boxes].bbox / im_scales[i]`
`164`	`164`	`# Predict the class label using the scores`
`165`		`- objects = torch.argmax(scores[keep_boxes][start_index:], dim=1)`
	`165`	`+ objects = torch.argmax(scores[keep_boxes][:, start_index:], dim=1)`
`166`	`166`
`167`	`167`	`info_list.append(`
`168`	`168`	`{`