search

cxliu0 / PET

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

For a single-image test. #15

Closed theseung closed 4 months ago

theseung commented 4 months ago

Hello! This model requires the use of a DataLoader for testing.

I want to test a single image and get the results, how can I do it?

Please help me.

cxliu0 commented 4 months ago

For single-image testing, you can discard the dataloader in engine.py, and directly load a single image following the data processing in SHA.py