采用keras深度学习框架搭建卷积神经网络模型实现垃圾分类,基于树莓派上进行实时视频流的垃圾识别源代码
简介:该垃圾分类项目主要在于对各种垃圾进行所属归类,本次项目采用keras深度学习框架搭建卷积神经网络模型实现图像分类,最终移植在树莓派上进行实时视频流的垃圾识别。前期:主要考虑PC端性能,并尽可能优化模型大小,训练可采用GPU,但调用模型测试的时候用CPU运行,测试帧率和准确性(测试10张左右图像的运行时间取平均值或实时视频流的帧率)。后期:部署在树莓派端,在本地进行USB摄像头实时视频流的垃圾分类(归类)。框架语言: keras+python。 PC端: Keras: 2.2.0 Opencv: 3.4 Python: 3.6 Numpy:1.16
文件列表
raspberrypi-Garbage-classification-master.zip
(预估有个17文件)
raspberrypi-Garbage-classification-master
Code2
AlexNet3_train.py
4KB
vgg16.py
2KB
remove_badimg.py
426B
data3_video_test.py
7KB
test.py
7KB
.keep
0B
mobilenet.py
2KB
readme.txt
176B
用户评论