君只见独不见的专栏

pytorch官方例程：DATA LOADING AND PROCESSING TUTORIALtorch.utils.data.Dataset 是dataset的抽象类，我们可以同过继承Dataset来定义自己的dataset，然后重写类里的两个方法：len 返回数据集的长度getitem 根据索引对数据集采样class FaceLandmarksDataset(Dataset):...

​在图像分割中，最后的输出结果mask结果是一个N x W x H的tensor，其中N表示有多少个类别，W是图像的宽度，H是图像的高度，每一个W x H都是一个二值矩阵第一步，转成numpy中的array第二步，N x W x H --> 1 x W x H（W x H），原理就是，对图像中的每一个像素，判断它在哪一类中的得分最高，然后把像素值置为得分最高类的序号。第三步，转成Im...