pytorch
君只见独不见
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
pytorch自定义数据集DataLoder
pytorch官方例程: DATA LOADING AND PROCESSING TUTORIAL torch.utils.data.Dataset 是dataset的抽象类,我们可以同过继承Dataset来定义自己的dataset,然后重写类里的两个方法: len 返回数据集的长度 getitem 根据索引对数据集采样 class FaceLandmarksDataset(Dataset):...原创 2019-06-22 19:49:38 · 925 阅读 · 0 评论 -
图像分割中mask的保存
在图像分割中,最后的输出结果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...原创 2019-06-22 20:17:06 · 5944 阅读 · 2 评论
分享