首页 > 其他分享 >在cmd下import cv2报错——OpenCV实现BRISK

在cmd下import cv2报错——OpenCV实现BRISK

时间:2022-11-30 18:07:55浏览次数:74  
标签:obj python BRISK cmd cv2 scene 报错 combo image


平台:win10 x64 +JetBrains PyCharm 2018.2.4 x64 +Anaconda3(python3.7.0+opencv3.4.5)

 
Issue说明:同学发了个python代码,想实现下brisk,帮同学解决,自己试验了下,但是报错ImportError:检查opencv的安装

from os import path
import sys
import numpy as np
try:
import cv2
except ImportError:
print('Couldn\'t find opencv so trying to use the fallback' \
' cv2.pyd (only for windows).')
from _cv2_fallback import cv2


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Helper Functions
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
def polygon_area(vertices):
"""Calculate the area of the vertices described by the sequence of vertices.

Thanks to Darel Rex Finley: http://alienryderflex.com/polygon_area/
"""
area = 0.0
X = [float(vertex[0]) for vertex in vertices]
Y = [float(vertex[1]) for vertex in vertices]
j = len(vertices) - 1
for i in range(len(vertices)):
area += (X[j] + X[i]) * (Y[j] - Y[i])
j = i
return abs(area) / 2 # abs in case it's negative


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Fundamental Parts
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# alternative detectors, descriptors, matchers, parameters ==> different results
detector = cv2.BRISK(thresh=10, octaves=1)
extractor = cv2.DescriptorExtractor_create('BRISK') # non-patented. Thank you!
matcher = cv2.BFMatcher(cv2.NORM_L2SQR)


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Object Features
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
obj_original = cv2.imread(path.join('source_images', 'object.png'),
cv2.CV_LOAD_IMAGE_COLOR)
if obj_original is None:
print ('Couldn\'t find the object image with the provided path.')
sys.exit()


# basic feature detection works in grayscale
obj = cv2.cvtColor(obj_original, cv2.COLOR_BGR2GRAY)
# mask with white in areas to consider, black in areas to ignore
obj_mask = cv2.imread(path.join('source_images', 'object_mask.png'),
cv2.CV_LOAD_IMAGE_GRAYSCALE)
if obj_mask is None:
print ('Couldn\'t find the object mask image with the provided path.' \
' Continuing without it.')


# keypoints are "interesting" points in an image:
obj_keypoints = detector.detect(obj, obj_mask)
# this lines up each keypoint with a mathematical description
obj_keypoints, obj_descriptors = extractor.compute(obj, obj_keypoints)
print ('Object Summary *************************************************')
print (' {} keypoints'.format(len(obj_keypoints)))


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Scene Features
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
scene_original = cv2.imread(path.join('source_images', 'scene.png'),
cv2.CV_LOAD_IMAGE_COLOR)
if scene_original is None:
print ('Couldn\'t find the scene image with the provided path.')
sys.exit()


scene = cv2.cvtColor(scene_original, cv2.COLOR_BGR2GRAY)
scene_mask = cv2.imread(path.join('source_images', 'scene_mask.png'),
cv2.CV_LOAD_IMAGE_GRAYSCALE)
if scene_mask is None:
print ('Couldn\'t find the scene mask image with the provided path.' \
' Continuing without it.')


scene_keypoints = detector.detect(scene, scene_mask)
scene_keypoints, scene_descriptors = extractor.compute(scene, scene_keypoints)
print ('Scene Summary **************************************************')
print (' {} keypoints'.format(len(scene_keypoints)))


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Match features between the object and scene
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
min_matches = 3
matches = matcher.match(obj_descriptors, scene_descriptors)
if len(matches) < min_matches:
print ('Not enough matches found between the image and scene keypoints.')
sys.exit()


#do some filtering of the matches to find the best ones
distances = [match.distance for match in matches]
min_dist = min(distances)
avg_dist = sum(distances) / len(distances)
# basically allow everything except awful outliers
# a lower number like 2 will exclude a lot of matches if that's what you need
min_multiplier_tolerance = 10
min_dist = min_dist or avg_dist * 1.0 / min_multiplier_tolerance
good_matches = [match for match in matches if
match.distance <= min_multiplier_tolerance * min_dist]
print ('Match Summary **************************************************')
print (' {} / {} good / total matches'.format(len(good_matches),
len(matches)))
if len(good_matches) < min_matches:
print ('not enough good matches to continue')
sys.exit()


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Calculate the shape of the object discovered in the scene.
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# extract the positions of the good matches within the object and scene
obj_matched_points = np.array([obj_keypoints[match.queryIdx].pt
for match in good_matches])
scene_matched_points = np.array([scene_keypoints[match.trainIdx].pt
for match in good_matches])
# find the homography which describes how the object is oriented in the scene
# also gets a mask which identifies each match as an inlier or outlier
homography, homography_mask = cv2.findHomography(obj_matched_points,
scene_matched_points,
cv2.RANSAC, 2.0)
print ('Homography Summary **************************************************')
print (' {} / {} inliers / good matches'.format(np.sum(homography_mask),
len(homography_mask)))


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Extract sizes and coordinates
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
obj_h, obj_w = obj.shape[0:2]
scene_h, scene_w = scene.shape[0:2]
#corners: opencv uses (top left, top right, bottom right, bottom left)
obj_top_left = (0, 0)
obj_top_right = (obj_w, 0)
obj_bottom_right = (obj_w, obj_h)
obj_bottom_left = (0, obj_h)
object_corners_float = np.array([obj_top_left, obj_top_right,
obj_bottom_right, obj_bottom_left],
dtype=np.float32)
#corners of the object in the scene (I don't know about the reshaping)
obj_in_scene_corners_float =\
cv2.perspectiveTransform(object_corners_float.reshape(1, -1, 2),
homography).reshape(-1, 2)


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Visualize the matching results
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# create a combined image of the original object and the scene
blue = (255, 0, 0)
green = (0, 255, 0)
red = (0, 0, 255)
combo_image = np.zeros((max(scene_h, obj_h), scene_w + obj_w), np.uint8)
combo_image[0:obj_h, 0:obj_w] = obj # copy the obj into the combo image
combo_image[0:scene_h, obj_w:obj_w + scene_w] = scene # same for the scene
combo_image = cv2.cvtColor(combo_image, cv2.COLOR_GRAY2BGR) # color for output
# draw a polygon around the object in the scene
obj_in_scene_offset_corners_float = obj_in_scene_corners_float + (obj_w, 0)
cv2.polylines(combo_image, [np.int32(obj_in_scene_offset_corners_float)],
True, blue, 2)
# mark inlier and outlier matches
for (x1, y1), (x2, y2), inlier in zip(np.int32(obj_matched_points),
np.int32(scene_matched_points),
homography_mask):
if inlier:
#draw a line with circle ends for each inlier
cv2.line(combo_image, (x1, y1), (x2 + obj_w, y2), green)
cv2.circle(combo_image, (x1, y1), 4, green, 2)
cv2.circle(combo_image, (x2 + obj_w, y2), 4, green, 2)
else:
#draw a red x for outliers
r = 2
weight = 2
cv2.line(combo_image,
(x1 - r, y1 - r), (x1 + r, y1 + r), red, weight)
cv2.line(combo_image,
(x1 - r, y1 + r), (x1 + r, y1 - r), red, weight)
cv2.line(combo_image,
(x2 + obj_w - r, y2 - r), (x2 + obj_w + r, y2 + r),
red, weight)
cv2.line(combo_image,
(x2 + obj_w - r, y2 + r), (x2 + obj_w + r, y2 - r),
red, weight)


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Do a sanity check on the discovered object
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
use_extracted = True # keep track of whether the extraction works or not
#check for size of the discovered result versus the original
scale_tolerance = 0.7
obj_area = polygon_area(object_corners_float)
obj_in_scene_area = polygon_area(obj_in_scene_corners_float)
area_min_allowed = obj_area * (1 - scale_tolerance) ** 2
area_max_allowed = obj_area * (1 + scale_tolerance) ** 2
if not (area_min_allowed < obj_in_scene_area < area_max_allowed):
print ('A homography was found but it seems too large or' \
' too small for a real match.')
use_extracted = False


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Extract the object from the original scene
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#max/min the corners to disentangle any flipped, etc. projections
tops_bottoms = list(corner[1] for corner in obj_in_scene_corners_float)
lefts_rights = list(corner[0] for corner in obj_in_scene_corners_float)
#limit the boundaries to the scene boundaries
top = max(int(min(tops_bottoms)), 0)
bottom = min(int(max(tops_bottoms)), scene_h - 1)
left = max(int(min(lefts_rights)), 0)
right = min(int(max(lefts_rights)), scene_w - 1)
extracted = scene_original[top:bottom, left:right]


# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
# Display and save all the results
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
combo_path = path.join('.', 'output_images', 'match_visualization.png')
cv2.imwrite(combo_path, combo_image)
mini_combo_h, mini_combo_w = int(round(float(combo_image.shape[0])/2)),\
int(round(float(combo_image.shape[1])/2))
mini_combo = cv2.resize(combo_image, (mini_combo_w, mini_combo_h))
cv2.imshow('match visualization', mini_combo)
# only display/save extracted if the previous tests indicated it was realistic
if use_extracted:
extracted_path = path.join('.', 'output_images', 'extracted.png')
cv2.imwrite(extracted_path, extracted)
mini_extr_h, mini_extr_w = int(round(float(extracted.shape[0])/2)),\
int(round(float(extracted.shape[1])/2))
mini_extr = cv2.resize(extracted, (mini_extr_w, mini_extr_h))
cv2.imshow('extracted image', mini_extr)


cv2.waitKey()
cv2.destroyAllWindows()

python_brisk_demo-master

原因:在cmd下python->import cv2就开始报错ImportError:检查opencv的安装,看来还没运行代码,刚开始import就报错了。原来是opencv_python的问题。

怀疑及解决方案:

Issue1:怀疑是电脑上安装的python3.6.5与Anaconda中的python3.7.0冲突,pip安装时(pip install opencv_python-3.4.5+contrib-cp37-cp37m-win_amd64.whl)没有识别对应的python。
解决办法:去掉python3.6.5的系统环境变量(D:\ProgramFiles\Python36),卸载opencv_python,重新安装opencv_python(pip install opencv_python-3.4.5+contrib-cp37-cp37m-win_amd64.whl),报错依旧

 

Issue2:怀疑电脑上的opencv_python版本低或文件破坏。
解决办法:重新下载了opencv_python-3.4.7+contrib-cp37-cp37m-win_amd64.whl,重新安装opencv_python(pip install opencv_python-3.4.7+contrib-cp37-cp37m-win_amd64.whl),报错依旧

 

Issue3:win+r打开windows命令窗口输入python显示的是python3.7.0,而在D盘文件夹下文件——>打开Windows PowerShell——>以管理员身份打开Windows PowerShell输入python显示的是python3.6.5(而我在Issue1时已经去掉python3.6.5的系统环境变量)
解决办法:把(D:\ProgramFiles\Python36)备份好后,卸载python3.6.5,报错依旧

 

Issue4:应该还是python多版本的问题。
解决办法:依次检查系统环境变量,看是否哪个包含了opencv3.6.5(因为近期使用CMake编译过opencv工程),还是没有找到,

最后在wang的用户变量中找到(D:\ProgramFiles\Python36;D:\ProgramFiles\Python36\Scripts\)去掉后

 

Issue5:怀疑是Anaconda的问题
解决办法:卸载重装,在重装安装opencv_python(pip install opencv_python-3.4.5+contrib-cp37-cp37m-win_amd64.whl),还是报错依旧。

 

Issue6:还是python多版本的问题
解决办法:最后在wang的用户变量中找到(D:\ProgramFiles\Python36;D:\ProgramFiles\Python36\Scripts\)去掉后,重新卸载后安装opencv_python(pip install opencv_python-3.4.5+contrib-cp37-cp37m-win_amd64.whl)win+r打开windows命令窗口输入python显示的是python3.7.0,然后输入import cv2 报错更改为ImportError: numpy.core.multiarray failed to import

 

Issue7:ImportError: numpy.core.multiarray failed to import,百度解决,是numpy版本过低的问题。
解决办法:更新numpy,使用:pip install -U numpy就可以更新numpy版本了。但是实验室网络太差,更新不成功,下载whl,(地址参看我的另一篇博客:Python入门之第三方模块安装——​​javascript:void(0)​​),下载最新的与64版本(numpy-1.17.0+mkl-cp37-cp37m-win_amd64.whl)后,安装(pip install numpy-1.17.0+mkl-cp37-cp37m-win_amd64.whl),在cmd下import cv2不报错了,问题解决。

参看:ImportError: numpy.core.multiarray failed to import——​​javascript:void(0)​​

 

Issue8:Pycharm下测试?

import cv2 as cv

src = cv.imread("D:/Working/opencvimg/lena.jpg")
#cv.namedWindow("input image", cv.WINDOW_AUTOSIZE)
cv.imshow("imput image", src)
cv.waitKey(0)
cv.destroyAllWindows()
print('Hi,Python!')

CV_test

解决办法:在pycharm上新建python工程->新建New Python File->粘贴上面的代码(更改图片的路径和名称),测试成功。

 

Issue9:Pycharm无法识别Python已安装的模块,如cv2(OpenCV)模块
解决办法:Pycharm的菜单  File | Settings | Settings窗口 | Project:XXXX | Project Interpreter项 | 窗口右侧 齿轮按钮点击 | Show All... | 然后依据上图提示打开“Interpreter Paths”窗口
将已安装的Python路径下的 Lib/site-packages 这个路径添加到“Interpreter Paths”窗口项中,最后保存

参看:本人另一篇博客:在cmd下可以import cv2,而Pycharm报错:找不到cv2 ——​​javascript:void(0)​​

 

Issue10:测试brisk,报错:Traceback (most recent call last):
  File "D:/Working/PycharmProjects/python_brisk_demo-master/brisk_demo.py", line 35, in <module>
    extractor = cv2.DescriptorExtractor_create('BRISK')  # non-patented. Thank you!
AttributeError: module 'cv2.cv2' has no attribute 'DescriptorExtractor_create'
解决办法:网上说OpenCV3.4.3及之后版本下没有使用该算法的权限,所以重新下载安装

opencv-contrib-python     3.4.2.16
opencv-python         3.4.2.16

清华大学opencv Python库:​​https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/opencv-python/​

 

Issue11:安装完成仍报错:没有权限
解决办法:此算法没有权限,将下载代码时的_cv2_fallback文件夹下(__pycache__文件夹——__init__.cpython-37.pyc,__init__.py,cv2.pyd)替换C:\ProgramData\Anaconda3\Lib\site-packages\cv2下相应的文件,还要改名哦(一定要备份更改的文件,如果失败,可以恢复)!系统警告,点击“继续”。运行代码,报错依旧!

 

Issue12:操作完成仍报错:没有权限
解决办法:此算法没有权限,下载下面的BRISK算法,实现。更改路径和名称后,成功!

#!/usr/bin/env python
# _*_coding:utf-8 _*_
import cv2
# import numpy


def main():
img = cv2.imread("D:/Working/opencvimg/lena.jpg")
cv2.imshow('Input Image', img)
cv2.waitKey(0)

brisk = cv2.BRISK_create()
keypoints = brisk.detect(img, None)

# 必须要先初始化img2
img2 = img.copy()
img2 = cv2.drawKeypoints(img, keypoints,img2, color=(0, 255, 0))
cv2.imshow('Detected BRISK keypoints', img2)
cv2.waitKey(0)


if __name__ == '__main__':
main()

OpenCV_BRISK_Test

标签:obj,python,BRISK,cmd,cv2,scene,报错,combo,image
From: https://blog.51cto.com/u_15405812/5900067

相关文章