Python OpenCV：使用 cv2.line() 从浮点数转换为整数答案

【问题标题】：Python OpenCV: Conversion from float to int for using cv2.line()Python OpenCV：使用 cv2.line() 从浮点数转换为整数
【发布时间】：2016-11-10 10:33:52
【问题描述】：

我正在使用 Python 2.7 和 OpenCV 3.1.0 处理 optical flow tutorial of openCV，并且对 cv2.line() 的使用有疑问。这是带有突出显示的感兴趣部分的原始代码：

import numpy as np
import cv2

cap = cv2.VideoCapture('slow.flv')

# params for ShiTomasi corner detection
feature_params = dict( maxCorners = 100,
                       qualityLevel = 0.3,
                       minDistance = 7,
                       blockSize = 7 )

# Parameters for lucas kanade optical flow
lk_params = dict( winSize  = (15,15),
                  maxLevel = 2,
                  criteria = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 0.03))

# Create some random colors
color = np.random.randint(0,255,(100,3))

# Take first frame and find corners in it
ret, old_frame = cap.read()
old_gray = cv2.cvtColor(old_frame, cv2.COLOR_BGR2GRAY)
p0 = cv2.goodFeaturesToTrack(old_gray, mask = None, **feature_params)

# Create a mask image for drawing purposes
mask = np.zeros_like(old_frame)

while(1):
    ret,frame = cap.read()
    frame_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # calculate optical flow
    p1, st, err = cv2.calcOpticalFlowPyrLK(old_gray, frame_gray, p0, None, **lk_params)

    # Select good points
    good_new = p1[st==1]
    good_old = p0[st==1]

    ##################  IMPORTANT  ##################
    # draw the tracks
    for i,(new,old) in enumerate(zip(good_new,good_old)):
        a,b = new.ravel()
        c,d = old.ravel()
        mask = cv2.line(mask, (a,b),(c,d), color[i].tolist(), 2)
        frame = cv2.circle(frame,(a,b),5,color[i].tolist(),-1)
    ##################  IMPORTANT  ##################

    ###########  START insert code below  ###########
    # Mean-vector of camera movement
    ############  END insert code below  ############

    img = cv2.add(frame,mask)

    cv2.imshow('frame',img)
    k = cv2.waitKey(30) & 0xff
    if k == 27:
        break

    # Now update the previous frame and previous points
    old_gray = frame_gray.copy()
    p0 = good_new.reshape(-1,1,2)

cv2.destroyAllWindows()
cap.release()

在我的工作区中，变量 a、b、c 和 d 显示为 array scalar float32。所以我假设，它们需要转换为 int 元组才能执行 cv2.line() 或 cv2.circle()。

当我尝试使用 cv2.line() 添加代码时，我必须使用转换为 int（见下文），否则我会收到一条非常明确的消息：TypeError: integer argument expected, got float

    ###################### START added code
    ofvec = p1 - p0
    ofvec = np.mean(ofvec, 1) # Collapse the first dimension
    ofvec_cam = np.mean(ofvec,0) # mean of camera movement

    height, width = old_frame.shape[:2]
    x0 = np.int(width/2)
    y0 = np.int(height/2)
    pt_center = (x0, y0) 

    x = np.int( x0 - ofvec_cam[0].tolist() )
    y = np.int( y0 - ofvec_cam[1].tolist() )
    pt_ofvec_cam = (x, y)

    frame = cv2.line(frame, pt_center, pt_ofvec_cam, [0, 0, 255], 2)
    ###################### END added code

谁能向我解释这个区别？提前致谢，祝您有美好的一天！ AMTQ

【问题讨论】：

添加了版本信息
嘿，您是如何找到 slow.flv 的？可以分享一下直接链接吗？

标签： python python-2.7 opencv opencv3.0

【解决方案1】：

似乎 cv2.line() 不同地对待两种类型的浮点数：“标准”Python 浮点数和 numpy 浮点数。请参阅使用 Python 2.7 和 OpenCV 3.1.0 的最低工作示例：

import numpy as np, cv2
mask = np.zeros([10, 20, 3], dtype=np.uint8)
color = [0, 0, 0]

# Using Numpy
a = np.float32(12.34)
mask = cv2.line(mask, (a,a), (a,a), color)

# Using standard Python data type
b = 12.34
mask = cv2.line(mask, (b,b), (b,b), color)

万一a命令执行顺利，万一b我们发现上面提到的错误：

in <module> mask = cv2.line(mask, (b,b), (b,b), color)
TypeError: integer argument expected, got float`

关于最初的问题，我确认在 OpenCV 教程中，变量 a、b、c 和 d 都是 numpy-floats，而在添加的代码中，变量 x 和 y 在转换为 numpy-ints 之前是标准 Python 浮点数通过 np.int()。

备注

两种数据类型都提供了一个方法__int__()，它返回浮点数的整数值（另请参见difference between native int type and the numpy int types）。

我发现的唯一要提及的是关于 OpenCV 2.4.13 文档中方法 fromarray 的注释：

注意在新的 Python 包装器（cv2 模块）中不需要该函数，因为 cv2 可以处理 Numpy 数组（这是唯一支持的数组类型）。

在 OpenCV 3.1.0 的文档中，方法 fromarray 不再存在。

【讨论】：