PatrickZad / CDPS

Apache License 2.0
2 stars 0 forks source link

About torch version #1

Open YangJae96 opened 1 month ago

YangJae96 commented 1 month ago

Hi, thank you for great work!

Which version of python and pytorch did you use?

Could you upload the requirements.txt file for enviroment build up?!

Thanks in advance.

PatrickZad commented 1 month ago

My environment is configured with Python 3.7 and Pytorch 1.8.2. You may also need to install the corresponding detectron2, pre-built with Pytorch 1.8 and CUDA11.1 in my case (please refer to https://detectron2.readthedocs.io/en/latest/tutorials/install.html), which should automatically install most of the dependencies. If there still exists missing packages, I believe they can be manually installed.

YangJae96 commented 1 month ago

Thank you.

I built detectron and tried to run the code.

But I am still facing difficult error.

from psd2.data.samplers.apk_sampler_cuda import APKSamplerPrec, pre_assign File "./psd2/data/samplers/apk_sampler_cuda.py", line 14, in import assign_cost_cuda ImportError: libpython3.7m.so.1.0: cannot open shared object file: No such file or directory

Have you faced this kind of problem?

Also, is there another version of CDPS annotation file rather than COCO style? It's hard to load the dataset on other person search codes directly.

Thank.s.

PatrickZad commented 1 month ago

Hi. Sorry for this late reply. For the error, please check if the 'assign_cost_cuda/assign_cost_cuda.so' file is missing, or you need to rebuild it with https://github.com/PatrickZad/assign_cost_cuda_CDPS. For manipulation with the annotations, please refer to 'psd2/data/datasets/cdps.py'