caffe的python接口以opencv的方式读取视频,图片做分类时出错的解决

来源:互联网 发布:软件开发思路 编辑:程序博客网 时间:2024/06/06 10:38
import syssys.path.insert(0, r"/home/lsn/Software/SSD/caffe/python")import caffeimport timeimport numpy as npimport cv2modelFile = "deploy.prototxt"pretrained = "snapshot/VGG_VOC0712_SSD_300x300_iter_100000.caffemodel"reload(sys)sys.setdefaultencoding("utf-8")caffe.set_mode_gpu();net = caffe.Classifier(modelFile, pretrained, raw_scale=256)#这里的通道用默认的input_image = cv2.imread(file)input_image = input_image / 255.0 #这里需要除以255做归一化,归一化后的图还可以像之前的方式使用net.predict([input_image], False)cv2.imshow("abc", input_image)key = cv2.waitKey()


    其实caffe有读图片的接口,也很方便,不过我好像没找到读视频的接口,所以就用opencv读了,然后用就报错了。原因是:
1、Mat本身用的是BGR的顺序存的数据。
2、caffe读图时会做归一化,所以得按它的方式来。

1 0
原创粉丝点击