search

cha15yq / CUT

Segmentation assisted U-shaped multi-scale transformer for crowd counting
17 stars 2 forks source link

How can I test the model on a single image? #3

Closed TheSenPie closed 2 weeks ago

TheSenPie commented 1 year ago

Hi, how can I use your model to process single image? My use case is, I would provide a single image to the program and it would output me the count of people in that image. Maybe you have an existing script, that just does that?

For example, I would give it this picture: image

And it would output number 24.

Thanks.

cha15yq commented 1 year ago

Hi there, I didn't preserve the visualization code. However, you can achieve that by modifying the test.py.