Skip to content

Christian7755/Computer-Vision_Soccer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Computer-Vision_Soccer

Using Explainability Methods for an Object Detection Modell from Roboflow.

Research Question: Which visual aspects influence the model's decision in ball detection?

Inspired by: https://github.com/roboflow/sports

Dataset: https://universe.roboflow.com/roboflow-jvuqo/football-ball-detection-rejhg/

Modell: YOLOv8

Lime Analysis:

SHAP Analysis

  • Theorie: SHAPley-Values: Determining which pixels contribute the most to the model's decision.

Grad-CAM Analysis

  • I had to train the YOLOv8 because the weights are only available in the roboflow premium version
  • The required weights are in the folder models/football_gradcam/weights so the training can be skipped

Setup und Execution

  • python environment: >= 3.8 (and <3.13)
  • Roboflow: API-Key needed
  • for Configuration: see config.py
  • UNZIP needed for the download of the data set
  • Execution should be performed from the project root directory

Steps:

  • run the first cells of setup_model.ipynb
    • Installing Requirements
    • Setup of the Model
    • Download of the Dataset
  • Now you're free to run the other notebooks, from the project root

About

Using Explainability Methods for an Object Detection Modell from Roboflow.

Resources

License

Stars

0 stars

Watchers

0 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors