ZhangDY827 / MHAN

The code for paper Remote Sensing Image Super-Resolution via Mixed High-Order Attention Network
17 stars 2 forks source link

关于训练用的数据 #1

Open JackeyZhang1001 opened 3 years ago

JackeyZhang1001 commented 3 years ago

最近在学习超分辨的内容,想试着跑通您论文中的代码,但是在运行的时候完发现训练数据的处理没有在github的readme中描述,所以我想问一下你们对于AID数据集是怎么使用的。在阅读代码后发现dataset中有HR 和 LR 的路径,想问一下这里文件路径是如何分配的,LR是从HR如何降采样得到的。

JackeyZhang1001 commented 3 years ago

并且在main_x4.py中,第87行 for batch in testing_data_loader:这句标红,且在data_x4.py中未找到testing_data_loader,也未在训练文件中引用,所以也希望您能解答

ZhangDY827 commented 3 years ago

这里面从HR 到 LR 是用matlab 的imresize 函数实现的,具体可以参考这个代码 https://github.com/twtygqyy/pytorch-LapSRN/blob/master/Test/generate_test_mat.m

ZhangDY827 commented 3 years ago

main_x4.py 里面87行那个test函数其实用处不大,测试得到的PSNR和SSIM都是在matlab上计算得到的。python计算的和matlab计算出来的PSNR有些不一样。一般情况下都是在MATLAB上计算的。