In ProposalTargetCreator after calculating iou of rois with bboxes, gt_assignment are calculated according to
gt_assignment = iou.argmax(axis=1)
However the iou has shape(N,1) where N is the number of rois+bboxes
taking argmax along first axis gives each element as it's own maximum.
Shouldn't it be along 0th axis so that we get the index of maximum iou?
I am not sure if the iou shape I am getting is correct or if this is the expected operation.
In ProposalTargetCreator after calculating iou of rois with bboxes, gt_assignment are calculated according to gt_assignment = iou.argmax(axis=1) However the iou has shape(N,1) where N is the number of rois+bboxes taking argmax along first axis gives each element as it's own maximum. Shouldn't it be along 0th axis so that we get the index of maximum iou? I am not sure if the iou shape I am getting is correct or if this is the expected operation.