本模块中使用.h5格式的数据,所用需要用到python的内置包:h5py。
图片中是否是猫?下面来看看识别过程。
# 导入要用到的包
import numpy
import h5py
# 查看图片
import matplotlib.pyplot as plt
%matplotlib inline
# 读入数据
train_dataset = h5py.File(r'/data/train_catvnoncat.h5', 'r')
test_dataset = h5py.File(r'/data/test_catvnoncat.h5', 'r')
# 提取数据信息
train_data_org = train_dataset['train_set_x'][:]
train_labels_org = train_dataset['train_set_y'][:]
test_data_org = test_dataset['test_set_x'][:]
test_labels_org = test_dataset['test_set_y'][:]
# 查看图片
plt.imshow(train_data_org[148])
<