京东fast-reid运行visualize_result.py出现error：IndexError: invalid index to scalar variable

系统管理员 2023-02-25 14:25 24阅读 0赞

## 1.问题 ##

IndexError: invalid index to scalar variable

## 2.解决 ##

修改/fast-reid-master/fastreid/evaluation/rank.py

line182:
    def evaluate_rank(
            distmat,
            q_pids,
            g_pids,
            q_camids,
            g_camids,
            max_rank=50,
            use_metric_cuhk03=False,
            use_cython=True
    ):
    改为：
    def evaluate_rank(
            distmat,
            q_pids,
            g_pids,
            q_camids,
            g_camids,
            max_rank=50,
            use_metric_cuhk03=False,
            use_cython=False
    ):

![2020071311312710.png][]

更靠谱的修改fast-reid-master/fastreid/utils/visualizer.py line72附近:

if len(query_indices) != 1:
        ax.set_title('{}/{:.2f}/cam{}'.format(query_name, self.all_ap[q_idx], cam_id))
    else:
        ax.set_title('{}/{:.2f}/cam{}'.format(query_name, self.all_ap, cam_id))

运行下面命令之后：

python ./demo/visualize_result.py --config-file './configs/Market1501/AGW_R50.yml' --vis-label --dataset-name 'Market1501' --output 'logs/market1501/agw_R50/agw_market1501_vis' --opts MODEL.WEIGHTS "./logs/market1501/agw_R50/model_final.pth"

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1OTc1NDQ3_size_16_color_FFFFFF_t_70][]

## 参考 ##

[1.fast-reid/issues/105（my issue）][1.fast-reid_issues_105_my issue]

[2020071311312710.png]: /images/20230209/577c4123627840de8bc4d53bdc788dad.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM1OTc1NDQ3_size_16_color_FFFFFF_t_70]: /images/20230209/26da26e22c724a109396e049eeb4607a.png
[1.fast-reid_issues_105_my issue]: https://github.com/JDAI-CV/fast-reid/issues/105