diff --git a/intermediate_source/torchvision_tutorial.rst b/intermediate_source/torchvision_tutorial.rst index 42cae8c4b7f..c6166e1c5b6 100644 --- a/intermediate_source/torchvision_tutorial.rst +++ b/intermediate_source/torchvision_tutorial.rst @@ -151,7 +151,7 @@ As datapoints are :class:`torch.Tensor` subclasses, wrapped objects are also ten # there is only one class labels = torch.ones((num_objs,), dtype=torch.int64) - image_id = torch.tensor([idx]) + image_id = idx area = (boxes[:, 3] - boxes[:, 1]) * (boxes[:, 2] - boxes[:, 0]) # suppose all instances are not crowd iscrowd = torch.zeros((num_objs,), dtype=torch.int64)