ZFU123456

HDv3模型的复现

5

flow, mask, merged = self.flownet(torch.cat((imgs, gt), 1), scale=scale, training=training) **loss_l1** = (merged[2] - gt).abs().mean() **loss_smooth** = self.sobel(flow[2], flow[2]*0).mean() # loss_vgg = self.vgg(merged[2], gt) if training: self.optimG.zero_grad() **loss_G = loss_cons +...

Can you provide the train dataset and train code？

Can you provide the lammbda/RSITMD.yaml ?