MY code is as
sample = images[0].permute(1, 2, 0).cpu().numpy() boxes = outputs[0][\'boxes\'].data.cpu().numpy() scores = outputs[0][\'scores\'].data.