search

cxliu0 / PET

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

Can it run TRANCOS? #6

Closed sullivantee closed 6 months ago

sullivantee commented 7 months ago

As the title says, I want to use TRANCOS to train the model, If it can run TRANCOS, what file do I need to change? and what do I need to change?

cxliu0 commented 7 months ago

Our approach is applicable to other counting tasks, and you can train PET on the TRANCOS dataset.

You need to customize the dataset :

  1. Customize the dataloader for TRANCOS dataset. Please refer to SHA.py to define your own dataset.
  2. Define TRANCOS in datasets/init.py.
  3. Modify train.sh to train PET on TRANCOS