fix: Fixed mobilenet imports for latest torch/torchvision releases (#115

) * fix import * update requirements
pyronear · Mar 26, 2021 · db08df7 · db08df7
1 parent 88958e4
commit db08df7
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/pyrovision/models/mobilenet.py b/pyrovision/models/mobilenet.py
@@ -3,7 +3,7 @@
 # This program is licensed under the GNU Affero General Public License version 3.
 # See LICENSE or go to <https://www.gnu.org/licenses/agpl-3.0.txt> for full license details.
 
-from torchvision.models.mobilenet import MobileNetV2, model_urls as imagenet_urls
+from torchvision.models.mobilenetv2 import MobileNetV2, model_urls as imagenet_urls
 from torchvision.models.utils import load_state_dict_from_url
 from .utils import cnn_model
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 opencv-python>=3.4.5.20
 pandas>=0.25.2
-torch>=1.2.0
-torchvision>=0.4.0
+torch>=1.8.0
+torchvision>=0.9.0
 tqdm>=4.20.0
 requests>=2.20.0
 pylocron>=0.1.3
diff --git a/setup.py b/setup.py
@@ -39,8 +39,8 @@
 requirements = [
     'opencv-python>=3.4.5.20',
     'pandas>=0.25.2',
-    'torch>=1.2.0',
-    'torchvision>=0.4.0',
+    'torch>=1.8.0',
+    'torchvision>=0.9.0',
     'tqdm>=4.20.0',
     'requests>=2.20.0',
     'pylocron>=0.1.3',