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

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages