上面两节通过介绍了几种预处理方法,分别是 Resize and Crop 和 Normalization。在完成图像预处理之后,加上之前手动搭建的神经网络,其实我们就可以对图形进行推理识别了。
这一节我们使用自己手写的算法、自己搭建的神经网络,来第一次完成一张图像的识别。
下面对一些重要代码做一下说明。
if __name__ == '__main__':
pics = GetPicList()
module = Resnet()
for filename in pics:
print("Begin predice with " + filename)
pre_out = PreProcess(filename)
res = module.run(pre_out)
# find inference result
out_res = list(res)
max_value = max(out_res)
index = out_res.index(max_value)
print("\npredict picture: " + filename)
print(" max_value: " + str(max_value))
print(" index: " + str(index)