from matplotlib import pyplot as plt
import numpy as np
plt.rcParams.update({
'font.family': 'sans-serif',
'font.sans-serif': ['Times New Roman'],
})
labels = ['Fast\nR-CNN', 'Faster\nR-CNN', 'SSD', 'YOLOV3','YOLOV4','MobilenetV3(TF)','MobilenetV3(IR)']
first = [79.51, 82.81, 81.23,72.63, 77.79, 84.81,84.53]
second = [5, 10, 20,23, 17, 42,43]
x = np.arange(len(labels)) # x轴刻度标签位置
width = 0.25 # 柱子的宽度
# 计算每个柱子在x轴上的位置,保证x轴刻度标签居中
# x - width,x, x + width即每组数据在x轴上的位置
# 这里举例三个,若多个,往两边加减width即可
plt.bar(x - width, first, width, alpha = 0.8, label='mAP')
for X, Y in zip(x, first):
plt.text(X - 0.30, Y + 0.05, '%.2f' % Y, fontsize = 8, ha = 'center', va = 'bottom')
plt.bar(x, second, width, alpha = 0.8, label='FPS')
for X, Y in zip(x, second):
plt.text(X, Y + 0.05, '%.0f' % Y, fontsize = 8, ha = 'center', va = 'bottom')
#plt.ylabel('values')
#plt.title('Example2')
# x轴刻度标签位置不进行计算
plt.xticks(x, labels = labels)
plt.legend() #在右下角生成标签,还有center、upper,如果不设置,系统自动找到合适地方
plt.savefig("map+fps.png")
plt.savefig( "map+fps.svg")
plt.show()
效果图如下![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/301ba2a6e820a9abe78f0db1cef3aeaf.png)