tensor_boxes = torch.FloatTensor(boxes). tensor_targets = torch.FloatTensor(labels) ... target = torch.hstack((bboxes, labels.unsqueeze(1))) ...
確定! 回上一頁