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))