在本文中,我们将了解如何使用 OpenCV 执行各种任务,重点是如何使用它来应用机器学习。
首先,让我们从安装开始,你需要在你的环境中安装 OpenCV 库,你可以通过运行以下命令来完成此操作:
pipinstallopencv-python
或者
condainstall-cconda-forgeopencv
一旦安装了 OpenCV,就可以开始在 Python 代码中使用它。以下是如何读取图像文件并显示它的示例:
importcv2#readtheimageimage=cv2.imread("image.jpg")#displaytheimagecv2.imshow("Image",image)cv2.waitKey(0)cv2.destroyAllWindows()
OpenCV 还提供了广泛的图像处理功能。以下是如何将图像转换为灰度并显示它的示例:
importcv2#readtheimageimage=cv2.imread("image.jpg")#converttheimagetograyscalegray=cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)#displaytheimagecv2.imshow("GrayscaleImage",gray)cv2.waitKey(0)cv2.destroyAllWindows()
OpenCV 的另一个重要特性是它能够检测图像中的特征。
例如,你可以使用 OpenCV 的cv2.CascadeClassifier
类来检测图像中的人脸:
importcv2#readtheimageimage=cv2.imread("image.jpg")#createtheclassifierclassifier=cv2.CascadeClassifier("path_to_classifier_xml")#detectfacesfaces=classifier.detectMultiScale(image,scaleFactor=1.3,minNeighbors=5)#drawarectanglearoundthefacesfor(x,y,w,h)infaces:cv2.rectangle(image,(x,y),(x+w,y+h),(255,0,0),2)#displaytheimagecv2.imshow("Faces",image)cv2.waitKey(0)cv2.destroyAllWindows()
OpenCV 还提供了许多基于机器学习的功能,例如检测、识别和跟踪。例如,你可以使用cv2.ml
模块来训练和使用机器学习模型。
importcv2importnumpyasnp#createthefeatureandlabelvectorsfeatures=np.array([[1,2],[3,4],[5,6],[7,8]])labels=np.array([1,2,3,4])#createtheSVMmodelsvm=cv2.ml.SVM_create()svm.setType(cv2.ml.SVM_C_SVC)svm.setKernel(cv2.ml.SVM_LINEAR)svm.setC(1.0)#trainthemodelsvm.train(features,cv2.ml.ROW_SAMPLE,labels)#testthemodelonnewdatanew_data=np.array([[2,3],[4,5]])result=svm.predict(new_data)print(result[1])
在上面的示例中,我们使用cv2.ml
模块创建了一个 SVM 模型,设置了模型的参数,使用我们的特征和标签向量对其进行了训练,然后在新数据上对其进行了测试。
另一个例子是使用深度学习,你可以使用OpenCV的cv2.dnn
模块来加载和使用预训练的深度学习模型cv2.dnn.readNetFromCaffe
,这是一个基于Caffe的深度学习模型。
importcv2#readtheimageimage=cv2.imread("image.jpg")#loadthedeeplearningmodelnet=cv2.dnn.readNetFromCaffe("path_to_prototxt","path_to_caffe_model")#settheinputblobblob=cv2.dnn.blobFromImage(image,1.0,(224,224),(104,117,123))net.setInput(blob)#getthepredictionspredictions=net.forward()#displaythepredictionsprint(predictions)
在上面的示例中,我们使用cv2.dnn
模块加载了一个深度学习模型,设置了输入 blob,然后使用该模型对我们的图像进行预测。
这些是你如何将 OpenCV 用于计算机视觉和机器学习任务的几个示例。OpenCV 拥有广泛的工具和功能,是一个强大的库,可供数据科学家用于满足他们的计算机视觉和机器学习需求。
OpenCV 强大的功能集使其成为图像和视频处理和分析的优秀库,机器学习的集成使其功能更加强大。
更多高级示例
对象跟踪:OpenCV 提供了广泛的对象跟踪算法,可用于跟踪视频流中的对象。例如,你可以使用cv2.TrackerKCF_create()
函数创建一个 KCF(Kernelized Correlation Filters)跟踪器,然后使用它来跟踪视频流中的对象。这是一个例子:
importcv2#createthevideocaptureobjectcap=cv2.VideoCapture("video.mp4")#getthefirstframeret,frame=cap.read()#selecttheobjecttotrackbbox=cv2.selectROI(frame,False)#createtheKCFtrackertracker=cv2.TrackerKCF_create()tracker.init(frame,bbox)#startthetrackingloopwhileTrue:#getthenextframeret,frame=cap.read()#updatethetrackersuccess,bbox=tracker.update(frame)#checkifthetrackingfailedifnotsuccess:break#drawtheboundingboxcv2.rectangle(frame,(int(bbox[0]),int(bbox[1])),(int(bbox[0]+bbox[2]),int(bbox[1]+bbox[3])),(255,0,0),2)#showtheframecv2.imshow("Tracking",frame)#exitiftheuserpressesthe'q'keyifcv2.waitKey(1)&0xFF==ord("q"):break#releasethevideocaptureandclosethewindowcap.release()cv2.destroyAllWindows()
光流:OpenCV 提供了广泛的光流算法,可用于跟踪视频流中对象的运动。一种流行的算法是 Farneback 算法,可用于估计两帧之间的光流。以下是如何使用此算法可视化视频流中的光流的示例:
importcv2#createthevideocaptureobjectcap=cv2.VideoCapture("video.mp4")#getthefirstframeret,frame1=cap.read()gray1=cv2.cvtColor(frame1,cv2.COLOR_BGR2GRAY)#startthetrackingloopwhileTrue:#getthenextframeret,frame2=cap.read()gray2=cv2.cvtColor(frame2,cv2.COLOR_BGR2GRAY)#calculatetheopticalflowflow=cv2.calcOpticalFlowFarneback(gray1,gray2,None,0.5,3,15,3,5,1.2,0)#visualizetheopticalflowmag,ang=cv2.cartToPolar(flow[...,0],flow[...,1])hsv=np.zeros((gray1.shape[0],gray1.shape[1],3),dtype=np.float32)hsv[...,0]=ang*180/np.pi/2hsv[...,1]=255hsv[...,2]=c
使用 OpenCV 机器学习功能的另一个示例是使用预训练模型进行对象检测。一种流行的对象检测模型是 Single Shot MultiBox Detector (SSD),它是一种基于深度学习的模型,可以检测图像中的多个对象。
importcv2#readtheimageimage=cv2.imread("image.jpg")#readthepre-trainedmodelandconfigfilesnet=cv2.dnn.readNetFromCaffe("ssd.prototxt","ssd.caffemodel")#createa4Dblobfromtheimageblob=cv2.dnn.blobFromImage(image,1.0,(300,300),(104.0,177.0,123.0))#settheblobasinputtothemodelnet.setInput(blob)#getthedetectionsdetections=net.forward()#loopoverthedetectionsforiinrange(detections.shape[2]):#gettheconfidenceofthedetectionconfidence=detections[0,0,i,2]#filteroutweakdetectionsifconfidence>0.5:#getthecoordinatesofthedetectionbox=detections[0,0,i,3:7]*np.array([w,h,w,h])(startX,startY,endX,endY)=box.astype("int")#drawthedetectionontheimagecv2.rectangle(image,(startX,startY),(endX,endY),(0,0,255),2)#displaytheimagecv2.imshow("Objects",image)cv2.waitKey(0)cv2.destroyAllWindows()
在上面的示例中,我们使用cv2.dnn.readNetFromCaffe
加载 SSD 模型及其配置文件,从输入图像创建一个 blob,将 blob 设置为模型的输入,运行前向传播以获得检测,过滤掉弱检测,并绘制检测在图像上。
另一个例子是使用 OpenCV 的cv2.Tracker
类来跟踪视频中的对象。
importcv2#Readvideocap=cv2.VideoCapture("video.mp4")#Readthefirstframeret,frame=cap.read()#Definetheregionofinterest(RoI)roi=cv2.selectROI(frame)#Initializethetrackertracker=cv2.TrackerKCF_create()tracker.init(frame,roi)#LoopovertheframeswhileTrue:#Readthenextframeret,frame=cap.read()ifnotret:break#Updatethetrackersuccess,roi=tracker.update(frame)#DrawtheRoIifsuccess:(x,y,w,h)=[int(v)forvinroi]cv2.rectangle(frame,(x,y),(x+w,y+h),(0,255,0),2)#Showtheframecv2.imshow("Frame",frame)key=c
使用 OpenCV 的另一个高级示例是使用图像抠图技术使图像中的对象消失。图像抠图是估计图像中每个像素的不透明度的过程,它允许你将前景对象与背景分开。
下面是如何使用 OpenCV 的cv2.createBackgroundSubtractorMOG2
函数从图像中提取前景对象并使其消失的示例:
importcv2#Readtheimageimage=cv2.imread("image.jpg")#CreatethebackgroundsubtractorbgSubtractor=cv2.createBackgroundSubtractorMOG2()#ApplythebackgroundsubtractortotheimagefgMask=bgSubtractor.apply(image)#Useamorphologicaloperatortoremovenoisekernel=cv2.getStructuringElement(cv2.MORPH_ELLIPSE,(3,3))fgMask=cv2.morphologyEx(fgMask,cv2.MORPH_CLOSE,kernel)#InvertthemasktogetthebackgroundbgMask=cv2.bitwise_not(fgMask)#Usethemasktoextractthebackgroundandtheobjectbg=cv2.bitwise_and(image,image,mask=bgMask)fg=cv2.bitwise_and(image,image,mask=fgMask)#Settheobjectpixelstotransparentfg[fg>0]=(255,255,255,0)#Combinethebackgroundandthetransparentobjectresult=cv2.addWeighted(bg,1,fg,1,0)#Showtheresultcv2.imshow("ObjectDisappeared",result)cv2.waitKey(0)cv2.destroyAllWindows()
在这个例子中,我们使用 OpenCV 的cv2.createBackgroundSubtractorMOG2
函数创建了一个背景减法器,然后将其应用于图像以提取前景对象。
然后我们使用形态学运算符从掩码中去除噪声。之后,我们反转掩码以提取背景,并使用掩码提取背景和对象。最后,我们将对象像素设置为透明,并将背景和透明对象组合在一起,以创建带有消失对象的最终结果。
总之,OpenCV 是用于计算机视觉和机器学习任务的强大且广泛使用的库。它提供了广泛的图像和视频处理工具,包括特征检测、图像识别、对象跟踪和机器学习。
本文中提供的示例演示了使用 OpenCV 执行各种任务是多么容易,例如读取和显示图像、将图像转换为灰度、检测图像中的特征以及对象检测和图像抠图。
OpenCV 还提供了许多基于机器学习的功能,例如使用 cv2.ml 和 cv2.dnn 模块进行检测、识别和跟踪。借助 OpenCV,开发人员可以轻松地将计算机视觉和机器学习功能集成到他们的项目中,并为各个行业创建创新的解决方案。
☆ END ☆
如果看到这里,说明你喜欢这篇文章,请转发、点赞。微信搜索「uncle_pn」,欢迎添加小编微信「 woshicver」,每日朋友圈更新一篇高质量博文。
↓扫描二维码添加小编↓
如果觉得《通过实际示例学习计算机视觉和机器学习的基本技术》对你有帮助,请点赞、收藏,并留下你的观点哦!