wwwww_bw的博客

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...

在分割任务中，我们拿到的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 .

import torchimport PILfrom PIL import Imageimport numpy as np# 二class DealDataset(torch.utils.data.Dataset): def __init__(self,imagespath,labelspath,Transform = None): #1：所有...