论文名称 基于固定锚定立体注意力网络的光学遥感图像去云方法代码运行
论文代码 https://github.com/ningjin00/TASANet?tab=readme-ov-file
论文地址
1环境创建
模型环境给了这几个包,如果你自带环境 那就运行代码 提示缺哪个装哪个
python == 3.12 rasterio == 1.3.10 torch == 2.3.0 yaml == 6.0.1 torchinfo == 1.8.0 torchmetrics == 1.3.2
2数据下载
在他的readme.md中找到 here ,点击here下载数据,数据500M大小。
6. Usage
If you want to continue training the model, you can run the model_train.py file for training.
If you want to test the model, you can run the model_predict.py file for testing and test model can be directly downloaded here.
3修改yaml文件路径
路径在TASANet-main\config\dataset_config.yaml,把其中的改为绝对路径
train_dataset_dir: "E:/qymx/TASANet-main/config/dataset_example/train_example" eval_dataset_dir: "E:/qymx/TASANet-main/config/dataset_example/eval_example"# 输入的验证文件路径 predict_dataset_dir: "E:/qymx/TASANet-main/config/dataset_example/test_example"
4点击训练
运行TASANet-main\model_train.py,开始训练