import matplotlib.pyplot as plt
import numpy as np
# Data for plotting
models = ['1', '2', '3', '4'] #model name
metric = [0.561, 0.461, 0.69, 0.89] #metric
x = np.arange(len(models)) # the label locations
width = 0.35 # the width of the bars
fig, ax = plt.subplots()
rects1 = ax.bar(x, metric, width)
# Add some text for labels, title and custom x-axis tick labels, etc.
ax.set_ylabel('MOTA')
ax.set_title('Vehicle') #vehicle, pedestrian, ...
ax.set_xticks(x)
ax.set_xticklabels(models)
ax.legend()
# Attach a text label above each bar in *rects*, displaying its height.
def autolabel(rects):
for rect in rects:
height = rect.get_height()
ax.annotate('{}'.format(height),
xy=(rect.get_x() + rect.get_width() / 2, height),
xytext=(0, 3), # 3 points vertical offset
textcoords="offset points",
ha='center', va='bottom')
autolabel(rects1)
fig.tight_layout()
plt.show()
XY 그래프
import matplotlib.pyplot as plt
import numpy as np
# Data for plotting
models = ['AB3DMOT', '2', '3', '4']
mota_values = [0.561, 0.559, 0.546, 0.561]
id_switch_values = [0.13, 0.12, 0.07, 0.08]
# Define colors and markers for each model
colors = ['green', 'blue', 'orange', 'red']
markers = ['o', '^', 's', '*']
x_range = (0.540, 0.565) # For MOTA
y_range = (0.00, 0.200) # For IDs Switch
# Create scatter plot
fig, ax = plt.subplots()
for i in range(len(models)):
ax.scatter(mota_values[i], id_switch_values[i], alpha=1, c=colors[i], marker=markers[i], label=models[i])
# Customize the plot
ax.set_title('Vehicle')
ax.set_xlabel('MOTA')
ax.set_ylabel('IDs Switch')
# Set the range for the axes
ax.set_xlim(x_range)
ax.set_ylim(y_range)
ax.grid(True, linestyle='--')
# Annotate each point with its MOTA and ID Switch values
for i, model in enumerate(models):
ax.annotate(f"({mota_values[i]}, {id_switch_values[i]})",
(mota_values[i], id_switch_values[i]),
textcoords="offset points",
xytext=(0,-15),
ha='center')
# Add legend
ax.legend()
# Show the plot
plt.show()
