在Mac下,深度学习CNN库Overfeat的配置及使用

Overfeat的主页:http://cilvr.nyu.edu/doku.php?id=software:overfeat:start

1. 下载Overfeat的代码包:

Download OverFeat-v04.tgz 

2. 下载已经学习好的网络权重数据,大小两个网络的数据都有,共1G左右,解压后放在程序能够访问的目录。

3. 安装依赖,并编译。

# Install compiler
sudo port install g++ imagemagick cmake
cd src
# Install torch :
sh install.sh
# Build overfeat
make all
4. 调用图片分类接口:
进入编译生成的bin目录,对某一个图片返回前五的分类结果置信度,默认使用小网络,-l参数使用大网络:
./overfeat -n 5 ../../samples/bee.jpg
./overfeat -n 5 -l ../../samples/bee.jpg
5.调用提取图片特征接口:
./overfeat -f ../../samples/bee.jpg
如果要提取4096x1的特征向量,用小网络图片要crop&resize到231x231的大小尺度;用大网络的话,图片要crop&resize到221x221大小尺度。
将提取的特征存成文本文件:./overfeat -f indir/bee.jpg > outdir/fea.txt
6. 利用python接口批量提取图片库特征的脚本:
import os
from glob import glob
import time

input_dir = ' '
output_dir = ' '
overfeat = '../overfeat'
overfeat_data_dir = '../data/default/'

files = glob(input_dir + '*.jpg')
files.sort()

start = time.clock()
for f in files:
    print f
    basename = os.path.basename(f)
    cmd = '{} -l -f -d {} {} > {}{}.txt'.format(overfeat, overfeat_data_dir, f, output_dir, basename)
    print cmd
    os.system(cmd)

end = time.clock()
print (end-start)

该批量提取特征的程序耗时还是比较多,因为对每一个图片提取特征的系统命令cmd,都要分别load及初始化训练好的网络参数。
7. 只load并初始化一次网络参数的C++程序,还在调试中。。。
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 12
    评论
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值