Pytorch CIFAR10图像分类 ZFNet篇

Pytorch CIFAR10图像分类 ZFNet篇

文章目录

  • Pytorch CIFAR10图像分类 ZFNet篇
    • 4. 定义网络(ZFNet)
      • ZFNet介绍
      • ZFNet结构
      • 判断是否使用GPU
      • summary查看网络
      • 测试和定义网络
    • 5. 定义损失函数和优化器
    • 6. 训练及可视化(增加TensorBoard可视化)
      • 开始训练
      • 训练曲线可视化
    • 7. 测试
      • 查看准确率
      • 查看每一类的准确率
      • 抽样测试并可视化一部分结果
      • 8. 保存模型
    • 9. 预测
      • 读取本地图片进行预测

你可能感兴趣的:(CIFAR10图像分类,pytorch,分类,深度学习)