本文整理《PyTorch深度学习与企业级项目实战》中项目案例所使用的模型,方便大家在学习、研究深度学习过程中做训练使用,这些案例也适合作为课程论文、毕业论文的素材,值得收藏和推荐。
第6章 迁移学习花朵识别项目实战
花朵数据集
ResNet模型
第7章 垃圾分类识别项目实战
垃圾图片数据集
MnasNet模型
第8章 短期电力负荷预测项目实战
电力负荷数据,智能电表采集
1D-CNN(一维卷积神经网络)模型
第9章 空气质量识别分类与预测项目实战
空气质量与污染物对应关系数据,公开数据收集
主成分分析,聚类(Cluster)分析
第10章 手写数字识别项目实战
MNIST手写数字数据集
LeNet5模型
第11章 人脸识别及表情识别实战
LFW数据集
Dlib库人脸检测器、训练好的人脸关键点检测器
FaceNet人脸识别器
第12章 图像风格迁移项目实战
自己收集的图像数据集
VGG-16模型
第13章 基于GAN生成动漫人物图像项目实战
动漫人物图像数据集
循环生成式对抗网络CycleGAN模型
第14章 糖尿病预测项目实战
印第安人糖尿病数据集
LSTM-CNN模型
第15章 基于大语言模型的自然语言处理项目实战
谣言检测系统项目,微博头条新闻数据,Transformer模型
新闻文本分类项目,中文新闻分类数据集,GPT-2模型
第16章 猴痘病毒识别项目实战
猴痘病毒分类数据集
ResNet101模型
第17章 X光肺部感染识别项目实战
胸部X射线图像(Chest X-Ray Images )数据集
ResNet50模型
第18章 乐器声音音频识别项目实战
乐器声音音频数据集
LSTM模型