ModuleNotFoundError: No module named 'sklearn.utils.linear_assignment_'

DeepSort 논문 코드 구현 중 발생한 에러이다.

 

scikit-learn의 기능 중 linear_assignment_가 0.23버젼부터 사라져

그 이상의 버젼을 사용한다면 나타나는 에러이다.

 

sklearn.utils.linear_assignment_는 scipy.optimize .linear_sum_assignment 로 대체 가능하다!

 

또는 scikit-learn 버젼 자체를 0.22로 다운그레이드해도 실행가능하다

(이 경우 scikit-learn의 다른 기능을 사용하지 못하게 될 수도 있다.)

 

두 가지 해결방법 모두 아래 코드에 기록해 놓았다.

 

# scipy로 바꾸어 사용하는 경우
# import 하는 부분에 해당 코드 입력하면 다른 코드 수정 불필요
from scipy.optimize import linear_sum_assignment as linear_assignment


# scikit-learn 다운그레이드
pip install scikit-learn==0.22.2