(转)opencv 代替caffe.io.load_image

最新推荐文章于 2024-08-05 23:09:30 发布

转载最新推荐文章于 2024-08-05 23:09:30 发布 · 269 阅读

0 ·

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

原文链接：http://www.cnblogs.com/houjun/p/9816361.html

标签

#人工智能

收录于

本文详细介绍了如何使用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