python openCV中的人口普查转换 [英] Census transform in python openCV
问题描述
我开始在与stereovision相关的项目中使用openCV和python.我找到了有关使用openCV进行C ++中的人口普查转换的文档的本页. 链接
I am starting to work with openCV and python in a project related with stereovision. I found this page of documentation about the Census Transform in C++ with openCV. link
有人知道python实现是否有类似的功能吗?
Does anyone know if there are similar functions for python implementation?
(例如功能的cv2.name)
(e.g. cv2.nameofthefunction)
谢谢大家!
PM 2Ring的出色解决方案(再次感谢您)可以与openCV一起使用,但只需一点改动:而不是使用Image.open
the excellent solution of PM 2Ring (thank you again) can work with openCV with this little change: instead of using Image.open
img = cv2.imread(img.png)
#some minor changes I needed like select some ROI and store them in img2[j]
#then a for cycle in which I wrote
src_img = img2[j]
h, w = src_img.shape
与"size"命令相比,"shape"指令似乎可以切换w和h的顺序.然后,我粘贴了PM 2Ring的其余代码,效果很好
where the "shape" instruction seems to switch the order of w and h compared to "size" command. Then I paste the rest of the code of PM 2Ring and it worked wonderfully
推荐答案
我不使用openCV,也不知道是否有人口普查转换的现有实现.但是,使用Numpy实施起来很容易.
I don't use openCV, and I don't know if there's an existing implementation of the Census Transform. However, it's easy enough to implement using Numpy.
这是一个简单的演示,使用 PIL 来处理图像加载并将数组数据转换回图像.
Here's a simple demo that uses PIL to handle loading the image and converting the array data back to an image.
#!/usr/bin/env python
''' The Census Transform
Scan an 8 bit greyscale image with a 3x3 window
At each scan position create an 8 bit number by comparing the value
of the centre pixel in the 3x3 window with that of its 8 neighbours.
The bit is set to 1 if the outer pixel >= the centre pixel
See http://stackoverflow.com/questions/38265364/census-transform-in-python-opencv
Written by PM 2Ring 2016.07.09
'''
import numpy as np
from PIL import Image
iname = 'Glasses0S.png'
oname = 'Glasses0S_census.png'
#Get the source image
src_img = Image.open(iname)
src_img.show()
w, h = src_img.size
print('image size: %d x %d = %d' % (w, h, w * h))
print('image mode:', src_img.mode)
#Convert image to Numpy array
src_bytes = np.asarray(src_img)
#Initialize output array
census = np.zeros((h-2, w-2), dtype='uint8')
#centre pixels, which are offset by (1, 1)
cp = src_bytes[1:h-1, 1:w-1]
#offsets of non-central pixels
offsets = [(u, v) for v in range(3) for u in range(3) if not u == 1 == v]
#Do the pixel comparisons
for u,v in offsets:
census = (census << 1) | (src_bytes[v:v+h-2, u:u+w-2] >= cp)
#Convert transformed data to image
out_img = Image.fromarray(census)
out_img.show()
out_img.save(oname)
来源
输出
原始的全彩眼镜图像由Gilles Tran使用POV-Ray创建,并且已在公共领域使用.可以在维基百科上找到.
The original full-colour Glasses image was created by Gilles Tran using POV-Ray, and is in the public domain. It may be found on Wikipedia.
这篇关于python openCV中的人口普查转换的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!