xiaozirui GitHub avatar

resnet18-gradcam-cifar10

xiaozirui

一个使用 PyTorch 手动实现 ResNet-18 用于 CIFAR-10 图像分类的项目,包含训练、验证和指标可视化;Grad-CAM 功能已列入计划但尚未实现。

Stars

5

7 天增长

暂无数据

Fork 数

0

开放 Issue

0

开源协议

MIT

最近更新

2026-07-29

AI 仓库情报摘要
FR-AI / ANALYSIS

为什么值得关注

逐层手动构建残差块,不依赖 torchvision 预定义网络,并针对 CIFAR-10 的 32x32 输入调整了 Stem 层(3x3 卷积、步长 1、省略最大池化),适合学习 ResNet 内部结构。

适合谁使用

  • 学习卷积神经网络手动实现的深度学习初学者
  • 需要 CIFAR-10 分类基线的 PyTorch 用户
  • 希望扩展类激活图可视化功能的研究者

典型使用场景

  • 学习 ResNet-18 网络结构的教学实践
  • 从零开始在 CIFAR-10 上训练分类模型
  • 作为图像分类实验的基准对比
  • 为后续添加 Grad-CAM 等可解释性方法提供基础

项目优势

  • 完全手动实现,不依赖 torchvision,增强对结构的理解
  • 针对 32x32 小图像优化了 Stem 层设计
  • 自动选择 CUDA/MPS/CPU 设备并保存最佳模型权重
  • 提供训练与验证的损失、准确率曲线可视化

使用前须知

  • Grad-CAM 可视化功能尚在计划中,当前版本未实现
  • 仅支持 CIFAR-10 数据集,未适配其他数据
  • 训练超参数(batch 128、20 epoch、SGD+余弦退火)固定,可能需要调整
  • 无预训练权重,需从头训练

README 快速开始

使用说明

项目描述

PyTorch ResNet-18 implementation with Grad-CAM visualization on CIFAR-10

相关仓库与替代方案

根据分类、Topic 和编程语言匹配的相似项目。

lopopolo
精选
lopopolo GitHub avatar

harness-engineering

Harness Engineering is a methodology for improving coding agent outputs by carefully crafting the environment around them—providing curated context, tools, and executable constraints that encode an organization’s nonfunctional requirements and cumulative lessons.

AI 与机器学习AI 智能体
2,390
slvDev
精选
slvDev GitHub avatar

esp32-ai

A 28.9 million parameter language model runs on an $8 ESP32-S3 microcontroller entirely on-device, generating simple stories at about 9.5 tokens per second.

AI 与机器学习大语言模型
1,960
littledivy
精选
littledivy GitHub avatar

mimic

mimic captures traffic from any iOS or web app and automatically generates a Python client library that lets you call the app's API like a regular library.

AI 与机器学习
1,482