未验证 提交 c1bd420f 编写于 作者: R Rakhil Immidisetti 提交者: GitHub

Fix rgb -> bgr while image loading and displaying

cap.read() returns BGR image.
So rgb->bgr is only needed for scipy.misc.imread 

cv2.imshow expects BGR image for displaying properly(RGB)
Refer: http://stackoverflow.com/a/15074748/1134940
上级 7d106c97
...@@ -263,11 +263,12 @@ if __name__ == '__main__': ...@@ -263,11 +263,12 @@ if __name__ == '__main__':
im_file = os.path.join(args.image_dir, imglist[num_images]) im_file = os.path.join(args.image_dir, imglist[num_images])
# im = cv2.imread(im_file) # im = cv2.imread(im_file)
im_in = np.array(imread(im_file)) im_in = np.array(imread(im_file))
if len(im_in.shape) == 2: if len(im_in.shape) == 2:
im_in = im_in[:,:,np.newaxis] im_in = im_in[:,:,np.newaxis]
im_in = np.concatenate((im_in,im_in,im_in), axis=2) im_in = np.concatenate((im_in,im_in,im_in), axis=2)
# rgb -> bgr # rgb -> bgr
im = im_in[:,:,::-1] im_in = im_in[:,:,::-1]
im = im_in
blobs, im_scales = _get_image_blob(im) blobs, im_scales = _get_image_blob(im)
assert len(im_scales) == 1, "Only single-image batch implemented" assert len(im_scales) == 1, "Only single-image batch implemented"
...@@ -366,8 +367,7 @@ if __name__ == '__main__': ...@@ -366,8 +367,7 @@ if __name__ == '__main__':
result_path = os.path.join(args.image_dir, imglist[num_images][:-4] + "_det.jpg") result_path = os.path.join(args.image_dir, imglist[num_images][:-4] + "_det.jpg")
cv2.imwrite(result_path, im2show) cv2.imwrite(result_path, im2show)
else: else:
im2showRGB = cv2.cvtColor(im2show, cv2.COLOR_BGR2RGB) cv2.imshow("frame", im2show)
cv2.imshow("frame", im2showRGB)
total_toc = time.time() total_toc = time.time()
total_time = total_toc - total_tic total_time = total_toc - total_tic
frame_rate = 1 / total_time frame_rate = 1 / total_time
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册