Cifar10 图像显示
我原打算在网上借鉴一下,然后竟然没有找到简单的转化cifar10数据到图片的算法,算了,自己写一个
import os
import numpy as np
import tensorflow as tf
from tensorflow import keras
import matplotlib.pyplot as plt
tf.random.set_seed(22)
np.random.seed(22)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
assert tf.__version__.startswith('2.')
# 导入cifar10 data
(x_train, y_train), (x_test, y_test) = keras.datasets.cifar10.load_data()
x_train, x_test = x_train.astype(np.float32) / 255., x_test.astype(np.float32) / 255.
shape=x_train.shape
# 看第二张图片
plt.imshow(x_train[2,:,:,:])
plt.show()
嗯,fine,其实就这一句:
plt.imshow(x_train[2,:,:,:])
plt.show()