OpenMMLab の MMTracking でconfig、checkpointをダウンロードしようとするとエラーが出た

Object Tracking

公開日：2022年9月16日
最終更新日：2022年9月20日


2022年9月19日「mmtrack==0.14.0」が公開されてこのバグは修正されています。


エラーが出るPythonスクリプト

import os
from mim.commands.download import download
os.makedirs('models', exist_ok=True)
checkpoint_name = 'siamese_rpn_r50_20e_lasot'
checkpoint = download(package="mmtrack", configs=[checkpoint_name], dest_root="models")[0]

エラー内容

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "D:\python\openmmlab_works\mmtrack\trackingenv\lib\site-packages\mim\commands\download.py", line 80, in download
    model_info = get_model_info(
  File "D:\python\openmmlab_works\mmtrack\trackingenv\lib\site-packages\mim\commands\search.py", line 175, in get_model_info
    dataframe = convert2df(metadata)
  File "D:\python\openmmlab_works\mmtrack\trackingenv\lib\site-packages\mim\commands\search.py", line 401, in convert2df
    for key, value in name2collection[collection_name].items():
KeyError: 'SELSA-TemporalRoIAlign'

原因

pip install mmtrack

pipで「mmtrack」をインストールしたのがいけなかったようです。
おそらくpipでインストールされる「mmtrack==0.13.0」がバグを抱えているのでしょう。

解決策

公式サイト通りにインストールすると解決しました。

git clone https://github.com/open-mmlab/mmtracking.git
cd mmtracking
pip install -r requirements/build.txt
pip install -v -e .

最新のdev版ではバグは取り除かれているようです。

環境

2つの環境でこの現象が起きました。

Windows 11
Ubuntu 20.04 on WLS2