Skip to content

PLangsteiner/Bildverarbeitung-Projekt

Repository files navigation

Development of a deep learning model for crowd counting

How to use this code

1. for Training:

The training part is an updated an adjusted version of the Crowd Counting Framework by Gao et al. (https://github.com/gjy3035/C-3-Framework).

  • requirements: Python 3.X, Pytorch 2.2, other libs in requirements.txt
  • set variables in config.py: dataset (SHHA, SHHB or SHHM), exp path, training with or without GPU support, other hyperparameters
  • in datasets/SHHX/setting.py set path to data with labels
  • run train.py

2. to use the crowd counting application:

  • additional required libraries: PIL and customtkinter
  • change path to 4 models that will be compared in gui.py
  • start gui.py, select image in file system and see results

img.png

About

Bildverarbeitungs Projekt über crowd counting

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published