Spaces:
Build error
Build error
Commit
·
e72a8c5
1
Parent(s):
cdbd3dd
Update model_core.py
Browse files- model_core.py +3 -3
model_core.py
CHANGED
@@ -2,9 +2,9 @@ import torch
|
|
2 |
import torch.nn as nn
|
3 |
import torch.nn.functional as F
|
4 |
|
5 |
-
from
|
6 |
-
from
|
7 |
-
from
|
8 |
|
9 |
|
10 |
class SRMPixelAttention(nn.Module):
|
|
|
2 |
import torch.nn as nn
|
3 |
import torch.nn.functional as F
|
4 |
|
5 |
+
from attention import ChannelAttention, SpatialAttention, DualCrossModalAttention
|
6 |
+
from srm_conv import SRMConv2d_simple, SRMConv2d_Separate
|
7 |
+
from xception import TransferModel
|
8 |
|
9 |
|
10 |
class SRMPixelAttention(nn.Module):
|