leonelhs commited on
Commit
d253868
1 Parent(s): 617d6e1

Making pypi library and CLI tool

Browse files
Files changed (2) hide show
  1. networks/models.py +2 -2
  2. networks/resnet_GN_WS.py +1 -1
networks/models.py CHANGED
@@ -1,7 +1,7 @@
1
  import torch
2
  import torch.nn as nn
3
- from networks.resnet_GN_WS import ResNet
4
- import networks.layers_WS as L
5
 
6
 
7
  def build_model(weights):
 
1
  import torch
2
  import torch.nn as nn
3
+ from FBA_Matting.networks import ResNet
4
+ import FBA_Matting.networks.layers_WS as L
5
 
6
 
7
  def build_model(weights):
networks/resnet_GN_WS.py CHANGED
@@ -1,5 +1,5 @@
1
  import torch.nn as nn
2
- import networks.layers_WS as L
3
 
4
  __all__ = ['ResNet', 'l_resnet50']
5
 
 
1
  import torch.nn as nn
2
+ import FBA_Matting.networks.layers_WS as L
3
 
4
  __all__ = ['ResNet', 'l_resnet50']
5