Python 足球队的分类

Python 足球队的分类,python,opencv,scikit-learn,k-means,Python,Opencv,Scikit Learn,K Means,我想对足球比赛中的球队进行分类 我尝试的是首先通过yolov3模型检测玩家。一旦我得到了玩家的边界框,我会制作一个玩家裁剪,这基本上就是图像中检测到的玩家所在的部分。我将播放器的子图像传递给一个函数find\u-magnitive\u-color(image),该函数应该返回该子图像的最主要颜色。由于某种原因,它不能正常工作 def查找主色(img): #平均值=平均值(轴=0)。平均值(轴=0) 像素=np.float32(图像整形(-1,3)) n_颜色=5 #最后一个参数是ε。它是终止更新

我想对足球比赛中的球队进行分类

我尝试的是首先通过yolov3模型检测玩家。一旦我得到了玩家的边界框,我会制作一个玩家裁剪,这基本上就是图像中检测到的玩家所在的部分。我将播放器的子图像传递给一个函数
find\u-magnitive\u-color(image)
,该函数应该返回该子图像的最主要颜色。由于某种原因,它不能正常工作

def查找主色(img):
#平均值=平均值(轴=0)。平均值(轴=0)
像素=np.float32(图像整形(-1,3))
n_颜色=5
#最后一个参数是ε。它是终止更新之前质心必须移动的最小量
标准=(cv2.TERM\u标准\u EPS+cv2.TERM\u标准\u MAX\u ITER,500,.05)
flags=cv2.KMEANS\u RANDOM\u CENTERS
#使用kmeans从完整的像素值集确定n_颜色质心
_,标签,调色板=cv2.kmeans(
像素、n_颜色、无、标准、10、标志)
_,counts=np.unique(标签,返回\u counts=True)
#打印(计数)
显性=np.int0(调色板[np.argmax(计数)])
回报主导
我希望此函数返回一个BGR值列表,如果玩家穿着黄色球衣,则该值表示一些黄色阴影;如果玩家穿着白色球衣,则该值表示一些白色阴影