更新 mse.py

This commit is contained in:
liwenyun 2024-12-22 17:25:34 +08:00
parent e5d5883912
commit 23e9cc015e
1 changed files with 11 additions and 6 deletions

17
mse.py
View File

@ -1,5 +1,8 @@
import matplotlib.pyplot as plt
import numpy as np
import tikzplotlib
# plt.style.use("ggplot")
MCS = [ 'MI-FGSM', 'PGD', 'Adv-\nAttribute', 'Adv-\nDiffusion','TCA$^2$']
gold_medal = [ 54.82, 49.72, 50.12, 59.28,63.03]
@ -7,19 +10,21 @@ silver_medal = [ 31.06, 24.57, 18.63, 32.61,36.08]
x = np.arange(len(MCS))
print(x)
width = 0.35
width = 0.4
gold_x = x
silver_x = x + width
plt.bar(gold_x,gold_medal,width=width,color="royalblue", label='Face++')
plt.bar(silver_x,silver_medal,width=width,color="tomato", label='Aliyun')
# plt.bar(silver_x,silver_medal,width=width,color="silver")
plt.xticks(x+width, labels=MCS)
plt.xticks(x+width/2, labels=MCS)
for i in range(len(MCS)):
plt.text(gold_x[i],gold_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=8)
plt.text(silver_x[i],silver_medal[i], silver_medal[i],va="bottom",ha="center",fontsize=8)
plt.text(gold_x[i],gold_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=10)
plt.text(silver_x[i],silver_medal[i], silver_medal[i],va="bottom",ha="center",fontsize=10)
# plt.text(bronze_x[i],bronze_medal[i], gold_medal[i],va="bottom",ha="center",fontsize=8)
plt.legend(loc=2,frameon=False)
plt.ylabel('Mean Confidence Scores')
plt.legend(loc=2,frameon=False, fontsize=10)
plt.ylabel('Mean Confidence Scores', fontsize=10)
# tikzplotlib.save("test.tex")
plt.show()
# import matplotlib.pyplot as plt