《PanNet: A Deep Network Architecture for Pan-Sharpening》ICCV 2017

一、必要的包安装

推荐一个快速上手Pytorch教程 eat_pytorch_in_20_days

pip install opencv-python
pip install h5py
pip install tensorboard
pip install scipy
pip install torchsummary

二、代码精读

代码文件夹含义

./test_data/ 测试数据
./test_results/ 测试数据最后得到的结果
./train_logs/ 存放的日志的地方,可以用tensorboard方便的看training_loss 和 validate_loss/ 的变化趋势
./training_data/ 训练数据,格式有所变化为 .h5 格式,不过数据都是一样的 Nx64x64x8
./Weights/ 模型的权重(参数)
./data.py 把数据读进来
./main_test_single.py 得到模型(权重)之后,测试的文件
./main_train.py 训练文件
./model.py 专门存放网络
Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐