pytorch学习

1、Tensor 与tensorflow相同,在pytorch框架中,同样使用张量(tensor)来传输数据。我们可以对tensor的维度进行自定义,如果图像转化为tensor,那么一般来说我们需要将图像先转类型为numpy(此时数据维度为N x H x W X C,即数量x高x宽x通道数),然后再使用toTensor()将数据维度转变为N x C x H x W,因为在训练中,需要将通道数提到前面。同时,需要特别注意的是,在训练网络中,图像转为tensor后的数据类型一般会是float或是uint8(在处理遥感图像时,需要特别小心,避免直接将int16的类型直接转化造成最后结果与原图类型不一