From 0c9d979c2171804a216836ae515654e4ff4f7d03 Mon Sep 17 00:00:00 2001 From: ShqWW Date: Tue, 20 Aug 2024 00:32:41 +0800 Subject: [PATCH] update --- ...nchor_num_method - 副本.py:Zone.Identifier | 0 thsis_figure/speed_method.py | 54 +++++++++++++++++++ 2 files changed, 54 insertions(+) create mode 100644 thsis_figure/anchor_num_method - 副本.py:Zone.Identifier create mode 100644 thsis_figure/speed_method.py diff --git a/thsis_figure/anchor_num_method - 副本.py:Zone.Identifier b/thsis_figure/anchor_num_method - 副本.py:Zone.Identifier new file mode 100644 index 0000000..e69de29 diff --git a/thsis_figure/speed_method.py b/thsis_figure/speed_method.py new file mode 100644 index 0000000..3b1a602 --- /dev/null +++ b/thsis_figure/speed_method.py @@ -0,0 +1,54 @@ +import matplotlib.pyplot as plt +import matplotlib as mpl + +# 设置全局字体为 Times New Roman +mpl.rcParams['font.family'] = 'Times New Roman' +mpl.rcParams['font.serif'] = ['Times New Roman'] +mpl.rcParams['axes.titlesize'] = 14 +mpl.rcParams['axes.labelsize'] = 12 +mpl.rcParams['xtick.labelsize'] = 12 +mpl.rcParams['ytick.labelsize'] = 12 +mpl.rcParams['legend.fontsize'] = 12 +mark_size = 8 + +# 定义数据 +data = { + 'LaneATT (2021)': {'x': [3.23, 5.01, 23.67], 'y': [75.09, 76.68, 77.02], 'color': 'magenta', 'marker': 'H'}, + 'CLRNet (2022)': {'x': [7.37, 8.81, 9.31, 14.36], 'y': [79.58, 79.73, 80.47, 80.13], 'color': 'orange', 'marker': 'p'}, + 'CLRerNet (2023)': {'x': [7.37, 9.31, 14.36], 'y': [80.76, 81.12, 80.91], 'color': 'orangered', 'marker': 'p'}, + 'ADNet (2023)': {'x': [8.4, 10.67], 'y': [77.56, 78.94], 'color': 'green', 'marker': 'v'}, + 'SRLane (2024)': {'x': [3.12], 'y': [79.73], 'color': 'red', 'marker': '*'}, + 'UFLDv2 (2022)': {'x': [2.7, 4.6], 'y': [75, 76], 'color': 'purple', 'marker': '^'}, + 'PolarRCNN-NMS (ours)': {'x': [3.71, 4.97, 5.47, 6.14], 'y': [80.81, 80.92, 81.49, 81.34], 'color': 'blue', 'marker': 'o'}, + 'PolarRCNN (ours)': {'x': [4.77, 6.10, 6.54, 7.13], 'y': [80.81, 80.92, 81.49, 81.34], 'color': 'cyan', 'marker': 'o'}, + } + + + + +plt.xlim(0, 30) + +# 绘制数据点 +for label, props in data.items(): + plt.plot( + props['x'], props['y'], + alpha=0.8, + c=props['color'], + marker=props['marker'], + # edgecolors='w', + markersize = mark_size, + linewidth=1.2, + label=label + ) + +# 设置标题和标签 +plt.grid(True, linestyle='-', alpha=0.5) +plt.xlabel('Latency (ms) on NVIDIA A100') +plt.ylabel('F1-score (%)') + +# 添加图例,并调整图例中的标记大小 +legend = plt.legend(loc="upper right") +for handle in legend.legend_handles: + handle._sizes = [20] +plt.savefig('speed_method.png', dpi=300) +plt.show() \ No newline at end of file