Unsupervised Homography Estimation with Coplanarity-Aware GAN论文代码调试

在这里记录一下此论文代码调试过程中遇到的问题，以供大家参考。

此篇工作的代码开源在megvii-research/HomoGAN: This is the official implementation of HomoGAN, CVPR2022 (github.com)但是在具体运行过程中出现了一些问题，

一、原始数据处理

原作者的几篇论文使用的都是Deep Homography Dataset这个数据集，可以直接通过百度网盘进行下载，除了这个数据集，还有两个内容需要下载，这点从代码中也可以看出，

 

1、下载test_list.txt用来提供训练/测试数据，可以从BasesHomo/dataset at main · megvii-research/BasesHomo · GitHub中找到，这也是作者同组的工作，里面有对应的测试和训练text。

2、下载定量评估结果所需要的原始数据，也就是上述图片中pt文件夹中的内容，这个也可以从BasesHomo/dataset at main · megvii-research/BasesHomo · GitHub中找到，就是这个名为Coordinate-v2文件夹。下载的原始数据集中还有一个Corrdinate文件夹，不过如果想要复现作者的精度的话需要Coordinate-v2文件夹。

注意所下载的训练序列和npy文件需要保持一致，不然会出现找不到文件的问题。

二、代码部分

在文章的具体运行时，出现了部分模块发生循环引用的问题，导致出现错误，主要出现在model文件夹中，只需要将部分函数单纯拆分出来即可解决。

建议大家在2080ti的机器上运行，可以避免出现版本不兼容等问题。

最后附上我的复现结果，

 跟作者文中所给出的结果基本一致。

如果后续有问题，欢迎大家和我一起讨论，我的邮箱是595260329@qq.com