是怎么找到对应的class name的,原理是什么

来源:2-3 实战分类模型之数据读取与展示

慕前端0046168

2020-09-23

为什么你设置了class__name,图片就自动会对应上名字呢

写回答

1回答

正十七

2020-10-04

同学你好,我理解你说的是这段代码:

def show_imgs(n_rows, n_cols, x_data, y_data, class_names):
    assert len(x_data) == len(y_data)
    assert n_rows * n_cols < len(x_data)
    plt.figure(figsize = (n_cols * 1.4, n_rows * 1.6))
    for row in range(n_rows):
        for col in range(n_cols):
            index = n_cols * row + col 
            plt.subplot(n_rows, n_cols, index+1)
            plt.imshow(x_data[index], cmap="binary",
                       interpolation = 'nearest')
            plt.axis('off')
            plt.title(class_names[y_data[index]])
    plt.show()

从倒数第二行可以看到,图片的显示跟title是一一绑定的,而title则是从class_names里根据索引获取的。所以能一一对应上。


0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程