search

layumi / AICIty-reID-2020

:red_car: The 1st Place Submission to AICity Challenge 2020 re-id track (Baidu-UTS submission)
MIT License
449 stars 110 forks source link

cropped_aicity folder order??? #34

Open vokhidovhusan opened 3 years ago

vokhidovhusan commented 3 years ago

I am confused how to provide cropped_aicity folder. Should I make cropped images folder structure as we do with test_data? my test_data structure is: test_data/gallery/im_test and test_data/query/im_query

When I run fast_submit_baidu.py I am getting the following error: RuntimeError: CUDA out of memory. Tried to allocate 1.39 GiB (GPU 0; 7.80 GiB total capacity; 5.19 GiB already allocated; 174.31 MiB free; 5.59 GiB reserved in total by PyTorch)

@layumi Could you please give me some guidance about how to prepare test_dir and crop_dir to run fast_submit_baidu.py?

Thanks again for your great job.

layumi commented 3 years ago

Hi @martianvenusian I think you are right.

The error is due to the memory limitation of your GPU. You may use small batchsize during inference.