pytorch
sakumashirayuki
这个作者很懒,什么都没留下…
展开
-
pytorch的tensorboard使用方法
笔者使用torch版本:1.5.0载入库的写法为from torch.utils.tensorboard import SummaryWriter程序中创建tensorboard的语句为writer = SummaryWriter(comment=f'LR_{lr}_BS_{batch_size}_SCALE_{img_scale}')意为自动生成保存路径名称。注释(学习率,batchsize和图像尺寸)加在文件名后面。然后可往writer中添加loss等信息,具体写法网上也容易查到。但是这原创 2020-05-30 22:17:28 · 722 阅读 · 2 评论 -
pytorch二分类损失函数BCEWithLogitsLoss
一个正确的语义分割二分类损失函数的计算应该是如下这样的criterion = nn.BCEWithLogitsLoss(weight=None)masks_pred = net(imgs)loss = criterion(masks_pred, true_masks) #使用二分类交叉熵epoch_loss += loss.item()先前在网络最后输出时使用sigmoid,然后使用B...原创 2020-05-01 16:48:24 · 2892 阅读 · 0 评论 -
pytorch语义分割计算mIoU
版本:python3pred为模型预测的label,像素0表示背景,像素1表示类别1,像素2表示类别2,以此类推。target为groundtruth,这里读入格式为PIL image,格式不一样的请自行修改这里的n_classes是目标物类别数。比如,对于只有背景和一个检测物类别的二分类问题,n_classes=1因为pythonfor循环的range(a,b),范围其实为[a,b),所...原创 2020-04-27 20:06:51 · 5283 阅读 · 0 评论