Skip to content

Commit 6df85a6

Browse files
sauercrowdfmassa
authored andcommitted
Added size information for inception v3 (#719)
* Added size information for inception v3 Since inception v3 expects in contrast to the other models a size of 299x299x3 I added a small note in the docstring to point that out (addresses #560) * inception size info in note box; fixed line length * removed trailing whitespace
1 parent ee5b4e8 commit 6df85a6

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

torchvision/models/inception.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,10 @@ def inception_v3(pretrained=False, **kwargs):
1717
r"""Inception v3 model architecture from
1818
`"Rethinking the Inception Architecture for Computer Vision" <http://arxiv.org/abs/1512.00567>`_.
1919
20+
.. note::
21+
**Important**: In contrast to the other models the inception_v3 expects tensors with a size of
22+
299x299x3, so ensure your images are sized accordingly.
23+
2024
Args:
2125
pretrained (bool): If True, returns a model pre-trained on ImageNet
2226
"""

0 commit comments

Comments
 (0)