Can't evaluate KeyError: 'query_f'`

[zubairahmed-ai]

I am running the python evaluate_st.py --name market1501 but I receive the following error

Traceback (most recent call last): File "evaluate_st.py", line 149, in <module> query_feature = result['query_f'] KeyError: 'query_f'

when I print the result I get this

`{'header': b'MATLAB 5.0 MAT-file Platform: posix, Created on: Thu Jun 6 22:53:34 2019', 'version': '1.0', 'globals': [], 'distribution': array([[[0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[0.00527704, 0.00263852, 0.00791557, ..., 0. ,
0. , 0. ],
[0.06461538, 0.09230769, 0.07384615, ..., 0. ,
0. , 0. ],
...,
[0.00226757, 0.00453515, 0.00680272, ..., 0. ,
0. , 0. ],
[0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[0. , 0. , 0. , ..., 0. ,
0. , 0. ]],

   [[0.06578947, 0.02631579, 0.02631579, ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.08928571, 0.04464286, 0.07142857, ..., 0.        ,
     0.        , 0.        ],
    ...,
    [0.        , 0.00591716, 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ]],

   [[0.075     , 0.08214286, 0.07857143, ..., 0.        ,
     0.        , 0.        ],
    [0.02518892, 0.05793451, 0.04282116, ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    ...,
    [0.00208768, 0.        , 0.00417537, ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ]],`

   ...,

   [[0.11363636, 0.04545455, 0.11363636, ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.04761905, 0.07936508, ..., 0.        ,
     0.        , 0.        ],
    [0.02777778, 0.01388889, 0.        , ..., 0.        ,
     0.        , 0.        ],
    ...,
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ]],

   [[0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    ...,
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ]],

   [[0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    ...,
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ],
    [0.        , 0.        , 0.        , ..., 0.        ,
     0.        , 0.        ]]])}`

[Wanggcong]

Do you run this code to get a st model?

python3 gen_st_model_market.py --name ft_ResNet50_pcb_market_e --data_dir "/home/huangpg/st-reid/dataset/market_rename/"

[zubairahmed-ai]

No I used the pretrained model pytorch_result.mat

[zubairahmed-ai]

If you could please answer this it will help me run it

[Wanggcong]

All of the steps are shown in readme.md. If you do not want to re-train a model. I suggest you carefully follow these steps:

1. create a folder (suppose you name it A)
2. pull the two pretrained models into that folder. (i.e., pytorch_result2.mat for the st model and net_last.pth for net weights) .
3. extract features, test (appearance feature extraction)
   python3 test_st_market.py --PCB --gpu_ids 2 --name A --test_dir "/home/huangpg/st-reid/dataset/market_rename/"
   note: "/home/huangpg/st-reid/dataset/market_rename/" is your data path.
4. evaluate (joint metric, you can use your own visual feature or spatial-temporal streams)
   python3 evaluate_st.py --name A

[zubairahmed-ai]

Ok let me try them now as I dont want to re-train a model

[zubairahmed-ai]

For anyone running this on Windows replace line 147 in test_st_market.py with this filename = path.split('\\')[-1] and line 153 with this if label[0:2]=='-1' or label[0:2]=='-1_c': and save yourself some headache/time

[zubairahmed-ai]

@Wanggcong I am running out of memory on running step 3 above, I am not sure what this will output. If its going to create a model that I can use in step 4 then can you please provide that model for me to evaluate? I am just using this for eval at the moment so really need some quick help.

[Wanggcong]

You could use small batchsize for test if it occurs " out of memory". For example,

python3 test_st_market.py --PCB --gpu_ids 2 --name A --test_dir "/home/huangpg/st-reid/dataset/market_rename/" --batchsize 1

[Wanggcong]

Step 3 is not a model, it is used for feature extraction for images.

[zubairahmed-ai]

Nope still not working with batch sizes of 4,2 and 1
Can you please share a pretrained model? Also how can I use that in a video?

[Wanggcong]

Pretrained models are already released in readme.md. (both google drive or baidu). No other pretrained models! And the steps are clear. I think some errors are not about my code.

[Wanggcong]

If you want to apply this code to your scenario, I think you could re-train the st (spatial-temporal) model. Because different scenarios have different st relationships. Directly using this code for your scenario could not work.

[zubairahmed-ai]

@Wanggcong Makes sense but I dont have test data :(

[zubairahmed-ai]

@Wanggcong I downloaded net_last.pth and pytorch_result2.mat files and using them in evaluate_st.py but I am getting the error which is what I opened this issue with, can you please help?

[Wanggcong]

Without some important information, I afraid it is difficult for me to give you a good suggestion. You may try python -mpdb xxx.py for debugging the code and find the key problem. But this could be out of range of my code.