/camera-helmet-detection

Final work developed for the Topics in Image Processing subject at UFMS

Primary LanguageJupyter NotebookMIT LicenseMIT

camera-helmet-detection

Final work developed for the Topics in Image Processing subject at UFMS. This version uses the model RTMDet tiny.

Dependencies

  • python 3.11.6
  • mmdet 3.2.0
  • cv2 4.7.0

How to use the application

To use the model with your GPU (uses "cuda:0" as default):

python3 run.py cuda

To use the model with your CPU:

python3 run.py cpu