我需要将一个函数从opencv c ++切换到opencv python。 c ++版本是:(只是我遇到问题的部分)
map_x.at<float>(j,i) = pc.x;
map_y.at<float>(j,i) = pc.y;
remap(frame, unDistFrame, map_x, map_y, CV_INTER_LINEAR, 0, Scalar(0, 0, 0));
在python中,我有:
rows,cols,channels = frame.shape
map_x = np.array((rows,cols, channels), np.uint8) # (that is: height, width,numchannels)
map_y = np.array((rows,cols, channels), np.uint8)
frameUnDist = np.array((rows,cols, channels), np.uint8)
for i in xrange(rows):
for j in xrange(cols):
p1 = [i, j]
p1_a = sendToFunction(params, p1)
np.insert(map_x, p1_a[0])
np.insert(map_y, p1_a[1])
cv2.remap(frame, frameUndist, map_x, map_y, flags=CV_INTER_LINEAR+CV_WARP_FILL_OUTLIERS, fillval=(0, 0, 0, 0))
插入行显然是错误的,因为我没有指定插入值的位置。我该怎么做?
感谢。
答案 0 :(得分:1)
在python中,OpenCV Mats实际上是numpy数组,所以你可以使用map_x[j,i] = pc.x
答案 1 :(得分:0)
解决了,谢谢。我需要项目集。
map_x.itemset((j,i),p1_a[0])
map_y.itemset((j,i),p1_a[1])