opencv 代替caffe.io.load_image

最新推荐文章于 2021-03-26 18:37:38 发布

原创最新推荐文章于 2021-03-26 18:37:38 发布 · 1k 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#caffe #opencv

收录于

caffe 同时被 2 个专栏收录

7 篇文章

订阅专栏

opencv

4 篇文章

订阅专栏

本文介绍使用Caffe框架进行图像预处理的方法，包括调整图像大小、减去平均值等步骤，为深度学习模型输入做准备。

self.net.blobs['data'].reshape(1, 3, self.image_resize, self.image_resize)
#image = caffe.io.load_image(image_file)
image = cv2.imread(image_file)
image = cv2.resize(image, (300,300))
img = np.zeros((1,3,300,300))
img[0,0,:,:] = image[:,:,0]-104.0
img[0,1,:,:] = image[:,:,1]-113.0
img[0,2,:,:] = image[:,:,2]-127.0
self.net.blobs['data'].data[...] = img