Function plot_without_labels

project_utils/visualization_utils.py:320–385

(save_path, lowDWeights, labels, camid)

def plot_without_labels(save_path, lowDWeights, labels, camid):
    import matplotlib.pyplot as plt
    padding_rate = 0.1
    plt.cla()
    X, Y = lowDWeights[:, 0], lowDWeights[:, 1]
    unique = np.unique(labels)
    num_type_color = len(unique)
    cmap = get_cmap(num_type_color)
    colors = {l: cmap(i) for i, l in enumerate(unique)}
    colors = sns.color_palette("viridis", n_colors=num_type_color)

    for x, y, s, cam in zip(X, Y, labels, camid):
        if cam == 9:
            plt.text(x, y, ' ', fontsize=3,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle',  # 圆圈
                               ec=colors[s],  # 边框颜色
                               fc=colors[s]  # 填充颜色
                               ))
        elif cam == 8:
            plt.text(x, y, ' ', fontsize=3,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='square',  # 圆圈
                               ec=colors[s],  # 边框颜色
                               fc=colors[s]  # 填充颜色
                               ))
        elif cam == 7:
            plt.text(x, y, ' ', fontsize=5,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle',  # 圆圈
                               ec=colors[s],  # 边框颜色
                               fc=colors[s]  # 填充颜色
                               ))
        elif cam == 3 or cam == 6:
            plt.text(x, y, ' ', fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle',  # 圆圈
                               ec=colors[s],  # 边框颜色
                               fc=colors[s]  # 填充颜色
                               ))
        elif cam == 0:
            plt.text(x, y, ' ', fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='circle',  # 圆圈
                               ec=colors[s],  # 边框颜色
                               fc=colors[s]  # 填充颜色
                               ))
        else:
            plt.text(x, y, ' ', fontsize=1,
                     ha='center', va='center',  # 水平居中,垂直居中
                     bbox=dict(boxstyle='square',  #
                               ec=colors[s],  # 边框颜色
                               fc=colors[s]  # 填充颜色
                               ))
    plt.xlim(X.min() - padding_rate * X.min(), X.max() + padding_rate * X.max())
    plt.ylim(Y.min() - padding_rate * Y.min(), Y.max() + padding_rate * Y.max())
    # plt.xlim(X.min(), X.max())
    # plt.ylim(Y.min(), Y.max())