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

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

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

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

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

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

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

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

在CNNNet类后新增Net类:

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

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

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

添加参数总数统计和打印

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

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

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

关于卷积内部运行印象

相关推荐
GISer_Jing3 小时前
Nano Banana:AI图像生成与编辑新标杆
前端·javascript·人工智能
新元代码4 小时前
Function Calling的现状和未来的发展
人工智能
jinxinyuuuus4 小时前
订阅指挥中心:数据可移植性、Schema设计与用户数据主权
数据仓库·人工智能
ASS-ASH4 小时前
视觉语言大模型Qwen3-VL-8B-Instruct概述
人工智能·python·llm·多模态·qwen·视觉语言模型·vlm
Xy-unu4 小时前
[LLM]AIM: Adaptive Inference of Multi-Modal LLMs via Token Merging and Pruning
论文阅读·人工智能·算法·机器学习·transformer·论文笔记·剪枝
kangk124 小时前
统计学基础之概率(生物信息方向)
人工智能·算法·机器学习
再__努力1点4 小时前
【77】积分图像:快速计算矩形区域和核心逻辑
开发语言·图像处理·人工智能·python·算法·计算机视觉
福客AI智能客服5 小时前
露营装备行业智能 AI 客服:从 “售后救火” 到 “售前场景赋能” 的转型路径
人工智能
ccLianLian5 小时前
DINO系列
人工智能·计算机视觉
Hcoco_me5 小时前
LLM(Large Language Model)系统学习路线清单
人工智能·算法·自然语言处理·数据挖掘·聚类