是怎么找到对应的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里根据索引获取的。所以能一一对应上。
00
相似问题