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

发布时间:2024年01月10日

上面两节通过介绍了几种预处理方法,分别是 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)
文章来源:https://blog.csdn.net/dongtuoc/article/details/135514249
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。