视觉/图像重磅干货，第一时间送达
import mathimport cv2import cvzonefrom cvzone.ColorModule import ColorFinderimport numpy as np
cv2.VideoCapture() # grabbing frames from the video file `Videos/vid (3).mp4myColorFinder = ColorFinder(False)hsvVals = {'hmin': 8, 'smin': 96, 'vmin': 115, 'hmax': 14, 'smax': 255, 'vmax': 255}
imgContours, contours = cvzone.findContours(img, mask, minArea=500)if contours:posListX.append(contours[0]['center'][0])posListY.append(contours[0]['center'][1])
A, B, C = np.polyfit(posListX, posListY, 2)a = Ab = Bc = C - 590 # 590 represents the y-value threshold for the "basket" areax = int((-b - math.sqrt(b ** 2 - (4 * a * c))) / (2 * a))prediction = 330 < x < 430 # Basket zone
if prediction:cvzone.putTextRect(imgContours, "Basket", (50, 150), scale=5, thickness=5, colorR=(0, 200, 0), offset=20)else:cvzone.putTextRect(imgContours, "No Basket", (50, 150), scale=5, thickness=5, colorR=(0, 0, 200), offset=20)
—THE END—
下载1：Pytorch常用函数手册
欢迎加入CV学习交流微信群！
觉得有用，记得点个赞和在看
