pytorch初步学习（一）：数据读取

最近从tensorflow转向pytorch，感受到了动态调试的方便，也感受到了一些地方的不同。

所有实验都是基于uint16类型的单通道灰度图片。

一开始尝试用opencv中的cv.imread读取图片，发现会默认读8位数据。。。后来还是改用了skimage读取图片。一个小坑。

在tensorflow中：

利用append得到数组x_test [batchsize，width，hight]

x_test = x_test[:, :, :, np.newaxis]

# 占位符

x=tf.placeholder(tf.float32, shape=[None, w, h, 1], name='x')

# 送入网络tensor维度依次为：batchsize，width，hight，channel

在pytorch中：

arr = np.asarray(img, dtype="float32")

data_x[i, :, :, :] = arr

i += 1

data_y.append(int(item[0]))

data_x = torch.from_numpy(data_x)

data_y = torch.from_numpy(data_y)

dataset = dataf.TensorDataset(data_x, data_y)

loader = dataf.DataLoader(dataset, batch_size=batchsize, shuffle=True)

#　送入网络的tensor维度依次为：batchsize，channel, width，hight

在tensorflow中需要自己写一个minibatch函数控制训练，在pytorch中可以调用dataloader将数据变成torch需要的tensor形式，并且不需要额外写minibatch函数。

巴特西

pytorch初步学习（一）：数据读取

最新文章

热门文章