pip install-able

pyproject.toml

@@ -0,0 +1,21 @@
+# Copyright (c) 2024 Synthesia Limited - All Rights Reserved
+#
+# Unauthorized copying of this file, via any medium is strictly prohibited.
+# Proprietary and confidential.
+
+[build-system]
+requires = ["setuptools", "setuptools-scm"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "modnet"
+version = "0.0.1"
+
+requires-python = ">=3.10"
+
+dependencies = [
+  "torch",
+]
+
+[tool.setuptools.packages.find]
+where = ["src"]

src/modnet/models/backbones/wrapper.py

@@ -4,7 +4,7 @@ from functools import reduce
 import torch
 import torch.nn as nn
 
-from .mobilenetv2 import MobileNetV2
+from modnet.models.backbones.mobilenetv2 import MobileNetV2
 
 
 class BaseBackbone(nn.Module):

src/modnet/models/modnet.py

@@ -2,7 +2,7 @@ import torch
 import torch.nn as nn
 import torch.nn.functional as F
 
-from .backbones import SUPPORTED_BACKBONES
+from modnet.models.backbones import SUPPORTED_BACKBONES
 
 
 #------------------------------------------------------------------------------