代码目录结构
log --训练日志
model --模型存储目录
config.py --配置文件
data_loader.py --自定义数据加载
model.py --模型构建文件
pridect.py -- 预测模块
train.py -- 训练模块
数据来源
https://www.heywhale.com/mw/dataset/60d9bd7c056f570017c305ee/file http:
//vcc.szu.edu.cn/research/2017/RSCM.html
model 文件夹中的模型
model 文件夹中的文件太大,没法直接上传,放到了百度网盘里面
链接:https://pan.baidu.com/s/1bPoDG-CBvEIK2Nh9bl9EIQ 提取码:yyzw
数据集地址
链接:https://pan.baidu.com/s/1GAJQgwLigry3nDn0CFbzdA 提取码:f5dd --
来自百度网盘超级会员 V5 的分享