caffe檢測代碼中加入batchsize,實現更快測試


imgs = np.zeros((batch_size, 3, height, width))#創建多維度數組
image_last = cv2.imread(image_path)

image = cv2.resize(image_last, (width, height))
#####儘量不要用caffe封裝的transform模塊,這樣跟c++工程實現會有不同。
image= image.astype(np.float32)
emptyImage = cv2.resize(image, (width, height), interpolation=cv2.INTER_CUBIC)
emptyImage[:, :, 0] = mu[0] * scale
emptyImage[:, :, 1] = mu[1] * scale
emptyImage[:, :, 2] = mu[2] * scale
transformed_image = image -emptyImage

transformed_image = transformed_image.transpose((2, 0, 1))
#################################################################################
imgs[a] = np.array(transformed_image)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章