首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何解决“使用scipy.optimize.linear_sum_assignment代替”

如何解决“使用scipy.optimize.linear_sum_assignment代替”
EN

Stack Overflow用户
提问于 2019-08-06 05:53:03
回答 2查看 7.1K关注 0票数 8

我正在使用python脚本来检测人员。我的脚本中有以下一行:

代码语言:javascript
运行
复制
import time
import cv2 as cv
import glob
import argparse
import sys
import numpy as np
import os.path
from imutils.video import FPS
from collections import deque
from sklearn.utils.linear_assignment_ import linear_assignment

当我运行我的脚本时,我有以下几行代码:

代码语言:javascript
运行
复制
/home/user/.local/lib/python3.6/site-packages/sklearn/utils/linear_assignment_.py:127:
     DeprecationWarning: The linear_assignment function is deprecated in 0.21 and will be removed from 0.23. Use scipy.optimize.linear_sum_assignment instead.
     DeprecationWarning)

请告诉我如何解决这个问题。

EN

回答 2

Stack Overflow用户

发布于 2019-09-18 12:28:32

您需要将sklearn.utils.linear_assignment_.linear_assignment函数替换为scipy.optimize.linear_sum_assignment function

区别在于返回格式:linear_assignment()返回numpy数组,linear_sum_assignment()返回numpy数组的元组。通过在数组中转换linear_sum_assignment()的输出并将其转置,可以获得相同的输出。

您的脚本应该如下所示:

代码语言:javascript
运行
复制
import time
import cv2 as cv
import glob
import argparse
import sys
import numpy as np
import os.path
from imutils.video import FPS
from collections import deque
from scipy.optimize import linear_sum_assignment

#compute your cost matrix
indices = linear_sum_assignment(cost_matrix)  
indices = np.asarray(indices)
indices = np.transpose(indices)
票数 18
EN

Stack Overflow用户

发布于 2019-11-18 19:18:57

linear_assignment替换为linear_sum_assignment

代码语言:javascript
运行
复制
# from sklearn.utils.linear_assignment_ import linear_assignment
from scipy.optimize import linear_sum_assignment
cost = np.array([[4, 1, 3], [2, 0, 5], [3, 2, 2]])
# result = linear_assignment(cost)
result = linear_sum_assignment(cost)
result = np.array(list(zip(*result)))

assignment.html

票数 3
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/57369848

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档