Model mAP Graphs

This project includes a python script that creates graphs by reading data from CSV files of models trained with YOLO. The script generates graphs with mAP50 and mAP50-95 values using the CSV files that YOLO outputs at the end of training (results.csv) read from the YAML configuration file.

Installation

1. Clone the repository:

git clone https://github.com/yusufesatt/model-map-comparison.git
cd model-map-comparison

2. Create Virtualenv (Optional):

python -m venv mapcomp_env

# Ubuntu & MacOS
source mapcomp_env/bin/activate

# Windows
mapcomp_env/Scripts/activate

3. Install the required dependencies:

pip install pandas matplotlib pyyaml

Usage

1. YAML Configuration File: Edit config.yaml and add the file path (file_path) and model name (model_name) for each model. A sample YAML file:

   - file_path: "path/to/model1.csv"
     model_name: "Model 1"
   - file_path: "path/to/model2.csv"
     model_name: "Model 2"
   

2. Running the Script: Run the following command in the terminal or command client:

   python main.py
   

   The script will generate the charts and display them on the screen.

Contributing

If you encounter issues or have suggestions for improvements, please report them on the GitHub repository 🚀.

License

This project is licensed under the MIT License.