File size: 530 Bytes
226675b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import torch
import torch.nn as nn
from rscd.models.backbones.lgpnet.BFE_DPN import BFExtractor
class LGPNet_a(nn.Module):
def __init__(self):
super().__init__()
self.backbone = BFExtractor(n_channels=3, n_classes=2)
def forward(self, xA, xB):
list = [] # 0: out1,1: out2,2: feat1,3: feat2
out1, feat1 = self.backbone(xA)
out2, feat2 = self.backbone(xB)
list.append(out1)
list.append(out2)
list.append(feat1)
list.append(feat2)
return list
|