def __init__(self, args, fourier): super(res_SF, self).__init__() self.conv = transfer_img(fourier) self.resnet = resnet50_F() self.cnn_c = cnn_C() self.sf = SF(args, (24, 8, 8), 2)
def __init__(self, args, fourier): super(resnet50, self).__init__() self.batch_size = args[0] self.cuda = args[1] cuda = self.cuda self.conv = transfer_img(fourier) self.resnet = resnet50_F() self.cnn_c = cnn_C() self.fc = FC(24, 8, 8, 2, cuda)
def __init__(self, args, fourier): super(res_TN, self).__init__() self.batch_size = args[0] self.cuda = args[1] b = self.batch_size cuda = self.cuda self.conv = transfer_img(fourier) self.resnet = resnet50_F() self.cnn_c = cnn_C() self.tn = TN(b, 24, 8, 8, 2, cuda=cuda, method=1)
def __init__(self, args, fourier): super(res_test, self).__init__() self.conv = transfer_img(fourier) self.resnet = resnet50_F() self.tr = TR(args, (24, 8, 8), 64) self.sf = SF(args, (24, 8, 8), 64) self.cnn_c = cnn_C() self.final = nn.Sequential(nn.Linear(128, 64), nn.ReLU(), nn.Linear(64, 32), nn.ReLU(), nn.Linear(32, 2))