boopathiraj commited on
Commit
dba0c30
·
verified ·
1 Parent(s): 589a0da

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modnet.py +10 -1
modnet.py CHANGED
@@ -3,7 +3,16 @@ import torch.nn as nn
3
  import torch.nn.functional as F
4
  import os
5
  from .wrapper import *
6
- import .SUPPORTED_BACKBONES
 
 
 
 
 
 
 
 
 
7
 
8
  #------------------------------------------------------------------------------
9
  # MODNet Basic Modules
 
3
  import torch.nn.functional as F
4
  import os
5
  from .wrapper import *
6
+
7
+
8
+ #------------------------------------------------------------------------------
9
+ # Replaceable Backbones
10
+ #------------------------------------------------------------------------------
11
+
12
+ SUPPORTED_BACKBONES = {
13
+ 'mobilenetv2': MobileNetV2Backbone,
14
+ }
15
+
16
 
17
  #------------------------------------------------------------------------------
18
  # MODNet Basic Modules