Confused regarding iou.argmax(axis=1)

[mkakade92]

trafficstars

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.