我正在YOLO网络上进行物体检测。处理图像时出现意外结果。我的发现是:
在使用cv2.rectangle()
方法在图像上绘制矩形时,在Spyder IDE上它将图像类型更改为NoneType
,而在Jupyter笔记本上则保持不变(即:numpy.ndarray
)。
代码如下:
img = cv2.imread('1527502132336.jpeg', 1)
result = tfnet.return_predict(img)
tl = (result[0]['topleft']['x'], result[0]['topleft']['y'])
br = (result[0]['bottomright']['x'], result[0]['bottomright']['y'])
label = result[0]['label']
img = cv2.rectangle(img, tl, br, (255, 0, 0), 3)
type(img)
我希望获得与Jupyter返回的结果相同的结果,以进行进一步处理。为什么我会出现这种奇怪的行为。