先下好资源及源代码,贴上论文地址
五类神奇宝贝样本在230左右,素材多样,包括卡通形象卡片玩具等。
工程的结构:
1.数据集包含五类。2.有6个检测图片来检测。3.包含smallerVGGNet模型类。还有其他文件,plot就是打出来的损失和准确率函数,lb.pickle是索引用的(不知),pokedex.model是模型文件用于存放各参数,train是训练程序,classify是测试程序。
还提供了安装Keras教程,以及在云测试。
使用的图像是96*96*3通道的图像。利用了dropout抑制过拟合,以及same pool来缩减空间维度。从3 x 3减少到2 x 2的最大池大小,以确保不会过快地减少空间维度。
将原始图片转成规定大小,并将图片转成数组:
image = cv2.resize(image