From 17520908a41c55732afa7696a7aa17eaf66dfc7d Mon Sep 17 00:00:00 2001 From: Keith Callenberg Date: Mon, 7 Feb 2022 14:48:14 -0500 Subject: [PATCH] Update links to torchvision docs --- detecto/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/detecto/utils.py b/detecto/utils.py index 896cc44..528b975 100644 --- a/detecto/utils.py +++ b/detecto/utils.py @@ -13,7 +13,7 @@ def default_transforms(): applied to images passed to classes in the :mod:`detecto.core` module. :return: A torchvision `transforms.Compose - `_ + `_ object containing a transforms.ToTensor object and the transforms.Normalize object returned by :func:`detecto.utils.normalize_transform`. @@ -83,7 +83,7 @@ def filter_top_predictions(labels, boxes, scores): def normalize_transform(): """Returns a torchvision `transforms.Normalize - `_ object + `_ object with default mean and standard deviation values as required by PyTorch's pre-trained models.