cxliu0 / PET

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

Can I evaluate some videos by using pre-trained model ? #14

Closed SHAIIM04 closed 4 months ago

SHAIIM04 commented 5 months ago

Thanks for releasing your code and making good efforts.

I would ask you, Can I use a pre-trained model to evaluate some videos? I want to know the crowd count of persons for each frame.

cxliu0 commented 5 months ago

Sure. You can use the pre-trained model to estimate the number of people in each frame.

However, the counting performance may not be guaranteed, because there may exist a domain gap between the videos and the data that used to train the model.

SHAIIM04 commented 5 months ago

Thanks for your responses

Is there a code for this task can you provide it to me?

Best regrds

cxliu0 commented 5 months ago

This repository supports the evaluation of a single image. You can refer to eval.py and engine.py.

For videos, you need to rewrite the dataloader before evaluation.