1.算法
采用深度学习方法进行刀具识别;
2.项目来源
本项目基础是采用标准图像测试数据集MNIST以及江南大学项目数据集安检刀具数据集;
3.项目技术
本项目采用深度学习算法,融入比较新的深度学习技术:VGG、Resnet、Densenet、Data Augmentation、Attention等技术对算法进行优化;
4.项目运行
(1)本例中主要是对刀具进行识别,首先对数据进行预处理,可以观看 load_dataset 脚本,对数据进行处理;
(2)然后运行 knife_train 进行训练和测试;
(3)最后在测试集MNIST以及安检数据集的准确率均为1.0;
欢迎各位对此项目感兴趣的朋友批评指正!
anselorville / image-recognition Goto Github PK
View Code? Open in Web Editor NEWThis project forked from zw76859420/image-recognition
采用深度学习方法进行刀具识别。
License: GNU General Public License v3.0