From d0bb4fb3df31955cd2c3488f117fbca969f8d3aa Mon Sep 17 00:00:00 2001 From: Sasank Chilamkurthy Date: Wed, 6 May 2020 21:05:57 +0530 Subject: [PATCH 1/3] Add all the latest models to hubconf --- hubconf.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index f43f922e89c..b89a92c00bd 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,6 +1,7 @@ # Optional list of dependencies required by the package dependencies = ['torch'] +# classification from torchvision.models.alexnet import alexnet from torchvision.models.densenet import densenet121, densenet169, densenet201, densenet161 from torchvision.models.inception import inception_v3 @@ -8,7 +9,16 @@ resnext50_32x4d, resnext101_32x8d, wide_resnet50_2, wide_resnet101_2 from torchvision.models.squeezenet import squeezenet1_0, squeezenet1_1 from torchvision.models.vgg import vgg11, vgg13, vgg16, vgg19, vgg11_bn, vgg13_bn, vgg16_bn, vgg19_bn -from torchvision.models.segmentation import fcn_resnet101, deeplabv3_resnet101 from torchvision.models.googlenet import googlenet from torchvision.models.shufflenetv2 import shufflenet_v2_x0_5, shufflenet_v2_x1_0 from torchvision.models.mobilenet import mobilenet_v2 +from torchvision.models.mnasnet import mnasnet0_5, mnasnet0_75, mnasnet1_0, \ + mnasnet1_3 + +# segmentation +from torchvision.models.segmentation import fcn_resnet50, fcn_resnet101, \ + deeplabv3_resnet50, deeplabv3_resnet101 + +# object detection +from torchvision.models.detection import fasterrcnn_resnet50_fpn, \ + maskrcnn_resnet50_fpn, keypointrcnn_resnet50_fpn From acf83ddbbc4daa0041302df25d263b8e936acafb Mon Sep 17 00:00:00 2001 From: Sasank Chilamkurthy Date: Mon, 11 May 2020 18:07:23 +0530 Subject: [PATCH 2/3] remove detection models from hubconf --- hubconf.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/hubconf.py b/hubconf.py index b89a92c00bd..45c1b1bddce 100644 --- a/hubconf.py +++ b/hubconf.py @@ -19,6 +19,3 @@ from torchvision.models.segmentation import fcn_resnet50, fcn_resnet101, \ deeplabv3_resnet50, deeplabv3_resnet101 -# object detection -from torchvision.models.detection import fasterrcnn_resnet50_fpn, \ - maskrcnn_resnet50_fpn, keypointrcnn_resnet50_fpn From 86d891ec48481bb82dc06745444ae53c64b7cdb8 Mon Sep 17 00:00:00 2001 From: Sasank Chilamkurthy Date: Mon, 11 May 2020 18:09:29 +0530 Subject: [PATCH 3/3] fix link error --- hubconf.py | 1 - 1 file changed, 1 deletion(-) diff --git a/hubconf.py b/hubconf.py index 45c1b1bddce..79c22bd938b 100644 --- a/hubconf.py +++ b/hubconf.py @@ -18,4 +18,3 @@ # segmentation from torchvision.models.segmentation import fcn_resnet50, fcn_resnet101, \ deeplabv3_resnet50, deeplabv3_resnet101 -