From e8d5193a10a48def6417b4722a6c1736ddb5528d Mon Sep 17 00:00:00 2001 From: neso613 <31642462+neso613@users.noreply.github.com> Date: Mon, 9 May 2022 17:19:04 +0530 Subject: [PATCH] Update linear_assignment.py --- OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py b/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py index 178456c..96a8794 100644 --- a/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py +++ b/OneStage/yolo/deep_sort_yolov3/deep_sort/linear_assignment.py @@ -1,7 +1,7 @@ # vim: expandtab:ts=4:sw=4 from __future__ import absolute_import import numpy as np -from sklearn.utils.linear_assignment_ import linear_assignment +from scipy.optimize import linear_sum_assignment as linear_assignment from . import kalman_filter