lilujunai / model-metrics-plot

🔨🔨🔨(mmplot)used to draw line graphs of multiple index parameters such as algorithm accuracy and speed of multiple deep learning models.

model-metrics-plot

model-metrics-plot(mmplot)

English | 简体中文

---

😎 About

This project is developed based on libraries such as Pandas and Matplotlib, and can be used to draw line graphs of multiple index parameters such as algorithm accuracy and speed of multiple deep learning models.

---

🥰Result

plot

data/Pytorch_models_data.csv	data/PaddleYOLO_models_data.csv	data/MMYOLO_model_data.csv

---

🔨Usage

requirement

pip install matplotlib
pip install pandas

mmplot install

git clone [e-mail]:isLinXu/model-metrics-plot.git
cd model-metrics-plot

pip install -e .

run

python3 main.py

or use your custom data csv

 python3 main.py -c 'csv_path' -n 'figture_name' -p 'plot_type' -t 'title_name' -x 'xlabel_name' -y 'ylabel_name' -f font_size -g False -v 'value_type' -r 'colors' 

 python3 main.py -c data/model_data.csv -n 'plot.jpg'-p 'line' -t 'MS COCO Object Detection' -x 'PyTorch FP16 RTX3080(ms/img)' -y 'COCO Mask AP val' -f 10 -v 'mAP' -r '#0000FF'