【Pytorch实战教程】基于RML2018的融合网络分类 | LSTM网络和ResNet网络融合 | 含数据集的处理

文章目录

• 1. 数据处理部分
• • 1.1 代码
  • • 1.1.1 导包
    • 1.1.2 拆分原始数据（仅第一次执行一次）
    • 1.1.3 画星座图保存本地
    • 1.1.4 选择要用到的数据集
    • 1.1.5 加载图片数据
    • 1.1.6 数据预处理主程序
    • 1.1.7 测试读IQ数据和图片数据
    • 1.1.8 数据预处理-整体代码
  • 1.2 解释
• 2. 融合网络部分
• • 2.1 代码
  • • 2.1.1 导包
    • 2.1.2 画混淆矩阵子函数
    • 2.1.3 画准确率与损失曲线子函数
    • 2.1.4 画测试集的SNR和识别率图的子函数
    • 2.1.5 创建一个文件夹用于保存结果
    • 2.1.6 读取图像和IQ数据集的类
    • 2.1.7 定义融合LSTM和ResNet模型的类
    • 2.1.8 模型训练主程序
  • 3. 输出结果