65、python - 利用手写的网络,成功预测一张图片

上面两节通过介绍了几种预处理方法,分别是 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)

你可能感兴趣的:(CV视觉算法入门与调优,python,开发语言,深度学习,人工智能,机器学习)