内涵:caffe踩过的一些坑

1、caffe.io.load_image()与cv2.imread()的区别与联系

from __future__ import division
a=caffe.io.load_image("1.jpg")
b=cv2.imread("1.jpg")


则a=b/255

2、

transformed_image = self.transformer.preprocess('data', image)
self.net.blobs['data'].data[...] = transformed_image


用datetime.datetime.now()测了一下caffe中这两句话的耗时,我的天,20几个毫秒,抵得上前向耗费的时间了。

3、之前一直都是一次前向,一副图片,今天看到了一次前向,一个batch的图片的代码,记录下

bs = len(fnames)  # batch size
in_shape = net.blobs['data'].data.shape
in_shape[0] = bs # set new batch size
net.blobs['data'].reshape(*in_shape)
net.reshape()
for i, f in enumerate(fnames):
    img = Image.open(f)
    # scale all images to 256x256
    img = img.resize((256,256), PIL.Image.ANTIALIAS)
    img = numpy.array(img).astype(numpy.float32)

    transformed_image = transformer.preprocess('data', img)
    #print transformed_image.shape

    # put the image into i-th place in batch
    net.blobs['data'].data[i,:,:,:] = transformed_image   

# after reading all images into batch, forward once:
net.forward()

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

学弟

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值