Pytroch
wwwww_bw
这个作者很懒,什么都没留下…
展开
-
Pytorch 02 绘制acc loss 曲线
1、空白列表 acc = [ ] loss = [ ] 对于epoch来说,每epoch记录一次acc,和loss 2、设置y轴信息 :像列表中添加信息 steps =len(dataloader) acc.append(epoch_acc / steps) loss.append(epoch_loss / steps) 3、设定x轴信息 x1 = range(0:steps) x2 = range(0:steps) 4、绘制图像 plt.subplot(21...原创 2020-08-13 16:03:18 · 3299 阅读 · 0 评论 -
pytorch 01 关于分割任务中 onehot 编码转换的问题
在分割任务中,我们拿到的label通常是由数字类别组成的,但是在应用某些损失函数时,我们需要把label转换成 one—hot编码的形式。 例如:原始label维度 224*224*1(由数字0-2组成) ,为一个三类别的分割任务,在onehot编码后维度为 224*224*3,(可以看成3张224*224*1的切片)。 代码: 一:当维度为 N 1 * one-hot后 N C * def make_one_hot(input, num_classes): """Convert .原创 2020-07-28 19:23:56 · 1200 阅读 · 0 评论 -
Pytorch 01:语义分割自定义数据集的读取
import torch import PIL from PIL import Image import numpy as np # 二 class DealDataset(torch.utils.data.Dataset): def __init__(self,imagespath,labelspath,Transform = None): #1:所有...原创 2019-11-05 11:13:10 · 2332 阅读 · 1 评论