Skip to content

Commit

Permalink
Merge pull request #105 from keithcallenberg/patch-1
Browse files Browse the repository at this point in the history
Update links to torchvision docs
  • Loading branch information
alankbi authored Feb 9, 2022
2 parents 3150cbc + 1752090 commit c57b5fe
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions detecto/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def default_transforms():
applied to images passed to classes in the :mod:`detecto.core` module.
:return: A torchvision `transforms.Compose
<https://pytorch.org/docs/stable/torchvision/transforms.html>`_
<https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.Compose>`_
object containing a transforms.ToTensor object and the
transforms.Normalize object returned by
:func:`detecto.utils.normalize_transform`.
Expand Down Expand Up @@ -83,7 +83,7 @@ def filter_top_predictions(labels, boxes, scores):

def normalize_transform():
"""Returns a torchvision `transforms.Normalize
<https://pytorch.org/docs/stable/torchvision/transforms.html>`_ object
<https://pytorch.org/vision/stable/transforms.html#torchvision.transforms.Normalize>`_ object
with default mean and standard deviation values as required by PyTorch's
pre-trained models.
Expand Down

0 comments on commit c57b5fe

Please sign in to comment.