FMA-Net
FMA-Net copied to clipboard
what the requirement is for the number of input images
def forward(self, x, y=None):
# x: [B, 3, T, H, W]
# y: [B, 3, T, sH, sW]
result_dict = {}
F, KD, f_Y, f, anchor_D = self.degradation_learning_network(x)
May I ask what the requirement is for the number of input images? When inputting the tensor in this shape, errors will be reported in many steps in the middle.