![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytroch
wwwww_bw
这个作者很懒,什么都没留下…
展开
-
Pytorch 02 绘制acc loss 曲线
1、空白列表acc = [ ]loss = [ ]对于epoch来说,每epoch记录一次acc,和loss2、设置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 · 3375 阅读 · 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 · 1233 阅读 · 0 评论 -
Pytorch 01:语义分割自定义数据集的读取
import torchimport PILfrom PIL import Imageimport numpy as np# 二class DealDataset(torch.utils.data.Dataset): def __init__(self,imagespath,labelspath,Transform = None): #1:所有...原创 2019-11-05 11:13:10 · 2380 阅读 · 1 评论