2021
04-27
04-27
深度学习小工程练习之tensorflow垃圾分类详解
介绍这是一个基于深度学习的垃圾分类小工程,用深度残差网络构建软件架构使用深度残差网络resnet50作为基石,在后续添加需要的层以适应不同的分类任务模型的训练需要用生成器将数据集循环写入内存,同时图像增强以泛化模型使用不包含网络输出部分的resnet50权重文件进行迁移学习,只训练我们在5个stage后增加的层安装教程需要的第三方库主要有tensorflow1.x,keras,opencv,Pillow,scikit-learn,numpy安装...
继续阅读 >