README.zh-CN.md

model-metrics-plot

model-metrics-plot(mmplot)

English | 简体中文

---

😎 介绍

本项目基于Pandas、Matplotlib等库开发，可用于绘制多个深度学习模型的算法精度、速度等多个指标参数的折线图。

---

🥰结果

绘制结果

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

data/llm_eval_data.csv	data/llm_eval_data.csv

---

🔨用法

依赖安装

pip install matplotlib
pip install pandas

mmplot安装

git clone [email protected]:isLinXu/model-metrics-plot.git
cd model-metrics-plot

pip install -e .

使用

python3 main.py

或者，你可以使用自定义数据。

 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' 

line

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'

python3 main.py -c data/PaddleYOLO_extra_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'

bar

python3 main.py -c data/MMYOLO_model_data.csv -p bar

leida

python3 main.py -c data/mllm_acc_eval-csv1029.csv -p leida