models import updated
Browse files
SiamMask/models/siammask.py
CHANGED
@@ -7,7 +7,7 @@ import torch
|
|
7 |
import torch.nn as nn
|
8 |
import torch.nn.functional as F
|
9 |
from torch.autograd import Variable
|
10 |
-
from utils.anchors import Anchors
|
11 |
|
12 |
|
13 |
class SiamMask(nn.Module):
|
|
|
7 |
import torch.nn as nn
|
8 |
import torch.nn.functional as F
|
9 |
from torch.autograd import Variable
|
10 |
+
from SiamMask.utils.anchors import Anchors
|
11 |
|
12 |
|
13 |
class SiamMask(nn.Module):
|
SiamMask/models/siammask_sharp.py
CHANGED
@@ -7,7 +7,7 @@ import torch
|
|
7 |
import torch.nn as nn
|
8 |
import torch.nn.functional as F
|
9 |
from torch.autograd import Variable
|
10 |
-
from utils.anchors import Anchors
|
11 |
|
12 |
|
13 |
class SiamMask(nn.Module):
|
|
|
7 |
import torch.nn as nn
|
8 |
import torch.nn.functional as F
|
9 |
from torch.autograd import Variable
|
10 |
+
from SiamMask.utils.anchors import Anchors
|
11 |
|
12 |
|
13 |
class SiamMask(nn.Module):
|
SiamMask/models/siamrpn.py
CHANGED
@@ -6,9 +6,9 @@
|
|
6 |
import torch
|
7 |
import torch.nn as nn
|
8 |
import torch.nn.functional as F
|
9 |
-
from utils.bbox_helper import center2corner
|
10 |
from torch.autograd import Variable
|
11 |
-
from utils.anchors import Anchors
|
12 |
|
13 |
|
14 |
class SiamRPN(nn.Module):
|
|
|
6 |
import torch
|
7 |
import torch.nn as nn
|
8 |
import torch.nn.functional as F
|
9 |
+
from SiamMask.utils.bbox_helper import center2corner
|
10 |
from torch.autograd import Variable
|
11 |
+
from SiamMask.utils.anchors import Anchors
|
12 |
|
13 |
|
14 |
class SiamRPN(nn.Module):
|