import os
import matplotlib.pyplot as plt
import matplotlib.lines as mlines
default_color = ['r','g','b','c','m','y','orange','brown']
default_marker = ['*','o','s','v','X','*','.','P']
default_fontsize = 48
class CMC:
def __init__(self,cmc_dict, color=default_color, marker=default_marker, fontsize=default_fontsize):
self.color = color
self.marker = marker
self.cmc_dict = cmc_dict
self.fontsize = fontsize
self.font = {'family': 'DejaVu Sans', 'weight': 'normal', 'size': self.fontsize}
self.sizeX = 20
self.sizeY = 10
self.lw = 10
self.ms = 40
def plot(self, title, rank=20, xlabel='Rank', ylabel='Matching Rates (%)', show_grid=True):
fig, ax = plt.subplots(figsize=(self.sizeX, self.sizeY))
fig.suptitle(title)
x = list(range(0, rank+1, 5))
plt.ylim(0.8, 1.0)
plt.xlim(1, rank)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.xticks(x)
plt.grid(show_grid)
method_name = []
i = 0
for name in self.cmc_dict.keys():
if rank < len(self.cmc_dict[name]):
temp_cmc = self.cmc_dict[name][:rank]
r = list(range(1, rank+1))
else:
temp_cmc = self.cmc_dict[name]
r = list(range(1, len(temp_cmc)+1))
if name == list(self.cmc_dict.keys())[-1]:
#globals()[name] = mlines.Line2D(r, temp_cmc, color=self.color[0], marker=self.marker[0], label='{:.1f}% {}'.format(self.cmc_dict[name][0]*100, name))
globals()[name] = mlines.Line2D(r, temp_cmc, color=self.color[0], marker=self.marker[0], label='{}'.format(name), linewidth=self.lw, markersize=self.ms)
else:
#globals()[name] = mlines.Line2D(r, temp_cmc, color=self.color[i+1], marker=self.marker[i+1], label='{:.1f}% {}'.format(self.cmc_dict[name][0]*100, name))
globals()[name] = mlines.Line2D(r, temp_cmc, color=self.color[i+1], marker=self.marker[i+1], label='{}'.format(name), linewidth=self.lw, markersize=self.ms)
i = i+1
ax.add_line(globals()[name])
method_name.append(globals()[name])
plt.legend(handles=method_name)
plt.rc('xtick', labelsize=self.fontsize)
plt.rc('ytick', labelsize=self.fontsize)
plt.rc('font', **self.font)
plt.show()
def save(self, title, filename,
rank=20, xlabel='Rank',
ylabel='Matching Rates (%)', show_grid=True,
save_path=os.getcwd(), format='png', **kwargs):
fig, ax = plt.subplots(figsize=(self.sizeX, self.sizeY))
fig.suptitle(title)
x = list(range(0, rank+1, 5))
plt.ylim(0.8, 1.0)
plt.xlim(1, rank)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.xticks(x)
plt.grid(show_grid)
method_name = []
i = 0
for name in self.cmc_dict.keys():
if rank < len(self.cmc_dict[name]):
temp_cmc = self.cmc_dict[name][:rank]
r = list(range(1, rank+1))
else:
temp_cmc = self.cmc_dict[name]
r = list(range(1, len(temp_cmc)+1))
if name == list(self.cmc_dict.keys())[-1]:
#globals()[name] = mlines.Line2D(r, temp_cmc, color='r', marker='*', label='{:.1f}% {}'.format(self.cmc_dict[name][0]*100, name))
globals()[name] = mlines.Line2D(r, temp_cmc, color='r', marker='*', label='{}'.format(name), linewidth=self.lw, markersize=self.ms)
else:
#globals()[name] = mlines.Line2D(r, temp_cmc, color=self.color[i], marker=self.marker[i], label='{:.1f}% {}'.format(self.cmc_dict[name][0]*100, name))
globals()[name] = mlines.Line2D(r, temp_cmc, color=self.color[i], marker=self.marker[i], label='{}'.format(name), linewidth=self.lw, markersize=self.ms)
i = i+1
ax.add_line(globals()[name])
method_name.append(globals()[name])
plt.legend(handles=method_name)
plt.rc('xtick', labelsize=self.fontsize)
plt.rc('ytick', labelsize=self.fontsize)
plt.rc('font', **self.font)
fig.savefig(os.path.join(save_path, filename+'.'+format),
format=format,
bbox_inches='tight',
pad_inches = 0, **kwargs)
Datasets
Models
