Skip to content

Commit

Permalink
[PaddlePaddle Hackathon] add InceptionV3 (#36064)
Browse files Browse the repository at this point in the history
* add inceptionv3
Co-authored-by: Ainavo <ainavo@163.com>
Co-authored-by: pithygit <pyg20200403@163.com>
  • Loading branch information
SigureMo authored Oct 22, 2021
1 parent ed478a3 commit ff06df6
Show file tree
Hide file tree
Showing 5 changed files with 571 additions and 2 deletions.
2 changes: 1 addition & 1 deletion python/paddle/tests/test_pretrained_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ def infer(self, arch):
def test_models(self):
arches = [
'mobilenet_v1', 'mobilenet_v2', 'resnet18', 'vgg16', 'alexnet',
'resnext50_32x4d'
'resnext50_32x4d', 'inception_v3'
]
for arch in arches:
self.infer(arch)
Expand Down
3 changes: 3 additions & 0 deletions python/paddle/tests/test_vision_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,6 +91,9 @@ def test_resnext152_32x4d(self):
def test_resnext152_64x4d(self):
self.models_infer('resnext152_64x4d')

def test_inception_v3(self):
self.models_infer('inception_v3')

def test_vgg16_num_classes(self):
vgg16 = models.__dict__['vgg16'](pretrained=False, num_classes=10)

Expand Down
2 changes: 2 additions & 0 deletions python/paddle/vision/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,8 @@
from .models import resnext101_64x4d # noqa: F401
from .models import resnext152_32x4d # noqa: F401
from .models import resnext152_64x4d # noqa: F401
from .models import InceptionV3 # noqa: F401
from .models import inception_v3 # noqa: F401
from .transforms import BaseTransform # noqa: F401
from .transforms import Compose # noqa: F401
from .transforms import Resize # noqa: F401
Expand Down
6 changes: 5 additions & 1 deletion python/paddle/vision/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@
from .resnext import resnext101_64x4d # noqa: F401
from .resnext import resnext152_32x4d # noqa: F401
from .resnext import resnext152_64x4d # noqa: F401
from .inceptionv3 import InceptionV3 # noqa: F401
from .inceptionv3 import inception_v3 # noqa: F401

__all__ = [ #noqa
'ResNet',
Expand All @@ -63,5 +65,7 @@
'resnext101_32x4d',
'resnext101_64x4d',
'resnext152_32x4d',
'resnext152_64x4d'
'resnext152_64x4d',
'InceptionV3',
'inception_v3'
]
Loading

0 comments on commit ff06df6

Please sign in to comment.