
训练自己的卷积神经网络模型
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本项目专注于构建与训练个人化的卷积神经网络(CNN)模型,旨在探索深度学习技术在图像识别和处理中的应用潜力。通过优化CNN架构,以期实现高精度的图像分类与目标检测功能。
在5到6台机器上进行测试以确保绝对可用。将要测试的数据集按照类别分别放置在data/train目录下,在retrain.bat文件中修改retrain.py和inception_model的路径。每次训练前需要清空bottleneck中的内容,并且把待测图片放在images目录里。为了评估训练好的模型,还需要修改生成输出文件out的位置。
本项目使用的是Inception v3架构,支持自定义数据集进行模型训练。目前的数据集中包含相貌等级的信息,在完成训练后可以查看效果以确保准确性。此外,文档中还包含了各种注意事项,并且需要安装TensorFlow环境来运行该项目。
全部评论 (0)
还没有任何评论哟~


