
PyTorch CIFAR-10: 图像分类示例
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目展示了如何使用PyTorch框架在CIFAR-10数据集上进行图像分类任务。通过构建神经网络模型,实现对包含飞机、汽车等物体的小型彩色图像的自动识别与分类。
PyTorch图片分类:CIFAR-10
目录:
1. 基于PyTorch的CIFAR-10图像分类项目使用ResNet-18 CNN模型,受PyTorch模板项目和Train CIFAR10项目的启发。然而,此存储库已与原模板分离,以便专注于快速研究与发展高级功能,并不考虑向后兼容性问题。特别地,在处理自己的数据集时可以轻松利用dataloader模块。
2. 开发环境概述:
- 操作系统(OS):Ubuntu MATE 18.04.3 LTS (Bionic)
- 图形处理器单元(GPU):NVIDIA TITAN Xp
- GPU驱动程序:Nvidia-450.102.04
- CUDA工具包:CUDA 10.2
修改日期:2021年3月31日。
全部评论 (0)
还没有任何评论哟~


