ObjectDetection
ObjectDetection copied to clipboard
In examples, incorrect anchors preview
From anchors you have format cthw, but matplotlib takes tlhw, so you should at first change datasize format.
def cthw2tlhw(boxes):
tl = boxes[:,:2] - boxes[:,2:]/2
return torch.cat([tl,boxes[:,2:]],1)