基于PyTorch的CIFAR-10图像分类项目总结(2)

  1. 添加每个类别准确率计算

在整体准确率计算后新增功能:

初始化每个类别的正确预测计数器和总样本计数器

遍历测试集统计每个类别的预测结果

计算并打印10个类别的单独准确率

修正了原代码中的语法错误

  1. 增加全局平均池化网络架构

在CNNNet类后新增Net类:

使用全局平均池化替代全连接层

减少参数数量,防止过拟合

保持相同的卷积层结构但修改全连接部分

添加参数总数统计和打印

3.思考:卷积层的通道为什么会增加

注意:在输入图像到达卷积层后原本的RGB三通道与卷积后的超参数不相关

个人感觉卷积核就像高维度想低维度的收缩核心

关于卷积内部运行印象

相关推荐
汉知宝科技8 小时前
跨境电商品牌合规:出海企业商标管理的特殊挑战与数字化应对
大数据·人工智能
ai产品老杨8 小时前
架构师深剖:基于 Docker 容器化与边缘计算的 AI 视频管理平台——支持 GB28181/RTSP 多协议接入与全源码交付
人工智能·docker·边缘计算
IT_陈寒8 小时前
Python的os.path.join居然能这么坑?
前端·人工智能·后端
2401_832298108 小时前
突破跨平台孤岛,OpenClaw全域系统互联能力,构建企业统一数字操作系统
人工智能
物联网IoT小易9 小时前
AI企业园区技术架构思考:大模型如何进入物理世界运营场景?
人工智能·智慧园区·智慧园区解决方案·ai智慧园区·aiot平台·ai企业园区
陈天伟教授9 小时前
图解人工智能(55)人工智能应用-机器翻译
人工智能·自然语言处理·机器翻译
watersink9 小时前
PagedAttention论文深度解析
人工智能
羊羊一洋9 小时前
对讲机核心技术解析:色码、亚音、脱网
人工智能·语音识别
OpenCSG9 小时前
不止 AI 编程:CSGLite 在多应用场景中的效率提升案例分析
人工智能
实在智能RPA9 小时前
航空维修知识库构建方法:从RAG到Agent-native的架构演进与全栈工程实践
人工智能·ai·架构