From 9521981279a09c69965db8b870ecadf3b414b5e2 Mon Sep 17 00:00:00 2001 From: GdoongMathew Date: Thu, 16 Jan 2025 22:35:14 +0800 Subject: [PATCH] docs: add `transforms` in `OxfordIIITPet`. (fix #8812 ) --- torchvision/datasets/oxford_iiit_pet.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/torchvision/datasets/oxford_iiit_pet.py b/torchvision/datasets/oxford_iiit_pet.py index 1d6d990fdf9..e5d20d315d2 100644 --- a/torchvision/datasets/oxford_iiit_pet.py +++ b/torchvision/datasets/oxford_iiit_pet.py @@ -27,6 +27,8 @@ class OxfordIIITPet(VisionDataset): transform (callable, optional): A function/transform that takes in a PIL image and returns a transformed version. E.g, ``transforms.RandomCrop``. target_transform (callable, optional): A function/transform that takes in the target and transforms it. + transforms (callable, optional): A function/transform that takes input sample + and its target as entry and returns a transformed version. download (bool, optional): If True, downloads the dataset from the internet and puts it into ``root/oxford-iiit-pet``. If dataset is already downloaded, it is not downloaded again. """