arunm8489 · June 4, 2020 02:48
diff --git a/19 b/19
 #eg detection input dimension [1, 255, 13, 13]
 prediction = x.view(batch_size, bbox_attrs*num_anchors, grid_size*grid_size) # 1x255x169     
 prediction = prediction.transpose(1,2).contiguous() #1x169x255
 prediction = prediction.view(batch_size, grid_size*grid_size*num_anchors, bbox_attrs) #1x507x85
	#eg detection input dimension [1, 255, 13, 13]
	prediction = x.view(batch_size, bbox_attrsnum_anchors, grid_sizegrid_size) # 1x255x169
	prediction = prediction.transpose(1,2).contiguous() #1x169x255
	prediction = prediction.view(batch_size, grid_sizegrid_sizenum_anchors, bbox_attrs) #1x507x85