使用numpy实现caffe/AlexNet图像分类器的前向检测功能

小编 2026-06-26 阅读:906 评论:0
虽然卷积神经网络之前已存,但AlexNet网络的实现加快了卷积深度神经网络的发展,这一点让AlexNet居功至伟。与之前的一篇LeNet的前向检测的计算相比,AlexNet网络的前向计算要复杂得多,共有五个...

虽然卷积神经网络之前已存,但AlexNet网络的实现加快了卷积深度神经网络的发展,这一点让AlexNet居功至伟。与之前的一篇LeNet的前向检测的计算相比,AlexNet网络的前向计算要复杂得多,共有五个卷积层和三个全连接层。还有两个额外的Local Response Normalization的运算,以进一步增强神经网络的准确度,用numpy实现起来也颇费周折。本文提到的相关Python代码可在https://github.com/LinkYe/LearnDeep上获得。

首先,笔者缺少环境训练AlexNet,模型文件是caffe官网下载的:http://dl.caffe.berkeleyvision.org/bvlc_alexnet.caffemodel。执行alexnet-weight.py将该模型文件重写为NPZ文件,在创建AlexNetCM对象时读取它:

\"\"

读取的五个卷积核需要进行翻转,这一点在上一篇文章也提到过,这里使用了flipReverse()方法。之后为AlexNetCM类实现了卷积/池化/LRN等方法,最后着重是前向检测的运算过程: 

\"\"

 与LeNet的前向检测运算相比,AlexNet网络的运算的卷积有很多的不同。其一是卷积的过程可能会分成两组,如图中的ngroup=2,还有第一层卷积的步长为4,如图中的stride=4。另外,后面四个卷积层的模式是SAME,即输入的图像大小与卷积的输出大小是相同的,因此指定了padLen为非零。

为了测试我们的AlexNetCM类的功能,在命令行执行python3 -i caffe-AlexNet.py,会对一张图片调用AlexNetCM及caffe检测,对比结果;详细的实现可参见caffe-AlexNet.py的最后五十多行。下面是笔者对一张图片的检测结果: 

\"\" 

令人惊诧的是,结果完全相同!这很让人欣慰——不过,由于LRN的性能很低,笔者实现的前向检测需要几十秒的时间才能完成。在上图执行完成后,可以手动输入detect_img进行检测。至此通过“重复造车”,我们基于numpy实现了AlexNet网络的前向检测的运算,进一步加深了对卷积深度神经网络的理解。

 

版权声明

本文仅代表作者观点,不代表百度立场。
本文系作者授权百度百家发表,未经许可,不得转载。

热门文章
  • 机房智能化温湿度解决方式之POE供电以太网温湿度传感器

    机房智能化温湿度解决方式之POE供电以太网温湿度传感器
    机房智能化温湿度解决方式之POE供电以太网温湿度传感器 北京盈创力和电子科技有限公司 智能型TCP网口温湿度记录仪 北京IP网络温湿度记录仪厂家,北京盈创力和 北京智能型TCP网口温湿度记录仪IP网络温湿度记录仪是一种新型的基于TCP/IP协议双绞线以太网标准温湿度采集模块,利用它可以实现现场温度值、相对湿度值的采集,同时利用其自身的RJ45通信接口可以方便地和机房监控主机或交换机集线器进行联网。 工作于-40℃~85℃工业级带...
  • Sequential Monte Carlo Methods (SMC) 序列蒙特卡洛/粒子滤波/Bootstrap Filtering

    Sequential Monte Carlo Methods (SMC) 序列蒙特卡洛/粒子滤波/Bootstrap Filtering
    Problem Statement 我们考虑一个具有马尔可夫性质、非线性、非高斯的状态空间模型(State Space Model):对于一个时间序列上的观测结果{yt,t∈N}\\{ y_t , t \\in N \\}{yt​,t∈N},我们认为每个观测结果yty_tyt​的生成依赖于一个无法直接观察的隐变量xt∈{xt,t∈N}x_t \\in \\{x_t , t \\in N \\}xt​∈{xt​,t∈N},即:p(...
  • HTTP状态保持的原理

    HTTP状态保持的原理
    a)在用户登录之后,浏览器返回响应的时候会在响应中添加上cookieb)浏览器接收到cookie之后会自动保存c)当用户再次请求同一服务器中的其他网页的时候,浏览器会自动带上之前保存的cookied)服务接收到请求之后可以请 request 对象中取到cookie 判断当前用户是否登录  Http是无状态的,就是连接时数据互通,关闭后...
  • Hive 系统函数及示例

    Hive 系统函数及示例
    查看所有系统函数 show functions; 函数分类 内置函数【系统函数】 数学函数: floor、round、ceil、cos、log2等 字符串函数: length、reverse、trim、lower、get_json_object、repeat等 收集函数: size 转换函数: cast 日期函数: year、month、datediff、date、date_add等 条件函数: coalesce、case…w...
  • CSRF的原理和防范措施

    CSRF的原理和防范措施
    a)攻击原理:i.用户C访问正常网站A时进行登录,浏览器保存A的cookieii.用户C再访问攻击网站B,网站B上有某个隐藏的链接或者图片标签会自动请求网站A的URL地址,例如表单提交,传指定的参数iii.而攻击网站B在访问网站A的时候,浏览器会自动带上网站A的cookieiv.所以网站A在接收到请求之后可判断当前用户是登录状态,所以...
标签列表