search

cxliu0 / PET

[ICCV 2023] Point-Query Quadtree for Crowd Counting, Localization, and More
MIT License
61 stars 5 forks source link

关于NWPU数据处理的问题 #30

Open StarrySky1029 opened 1 month ago

StarrySky1029 commented 1 month ago

您好作者，很喜欢您的文章。我想复现一下关于NWPU数据集，想问一下能否提供一下像SHA、SHB那样的数据处理代码。并且对于定位的指标，需要对比两个txt文件，如何得到训练出的NWPU模型所得到的txt文件呢？

cxliu0 commented 1 month ago

Thanks for your interest in our work. For your questions:

  1. Regarding the NWPU dataset, you can refer to preprocess_dataset.py L135 for data processing.
  2. About the format of the txt file, please refer to the official guideline. You need to save the results following the format mentioned in the guideline. For your convenience, you can save the results in engine.py L146.