Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update for latest sklearn #220

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

yasutomo57jp
Copy link

No description provided.

@gndowns
Copy link

gndowns commented Dec 15, 2020

Seconding this.
Reference here:
https://stackoverflow.com/questions/62390517/no-module-named-sklearn-utils-linear-assignment

@danielcrane
Copy link

Thirding this, however I'd personally recommend changing it to be something more like the following to avoid the unnecessary np.vstack().T:

from scipy.optimize import linear_sum_assignment
   indices_rows, indices_cols = linear_sum_assignment(cost_matrix)

    matches, unmatched_tracks, unmatched_detections = [], [], []
    for col, detection_idx in enumerate(detection_indices):
        if col not in indices_cols:
            unmatched_detections.append(detection_idx)
    for row, track_idx in enumerate(track_indices):
        if row not in indices_rows:
            unmatched_tracks.append(track_idx)
    for row, col in zip(indices_rows, indices_cols):

jimtang123 added a commit to jimtang123/deep_sort that referenced this pull request Jul 22, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants