- 博客(3)
- 收藏
- 关注
原创 pytorch利用pos_weight解决分类不平衡问题
首先简单介绍损失函数,放两张之前的笔记,这里不再赘述,可直接跳过。下面重点介绍pos_weight的问题,直接上代码import torchimport torch.nn as nndevice = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")a=torch.tensor([1.0,2.0])m=nn.Sigmoid()b=torch.Tensor([0,1])loss_function = nn.BCEWi
2020-12-18 16:45:49 4123 2
翻译 弱监督论文解读:CADN- 用于表面缺陷检测的基于弱监督学习的分类感知对象检测网络
Article history:2020年1月26日收到,2020年6月17日修订,2020年7月31日接受,2020年8月1日网络上线Keywords:Weakly supervised learning ,Automated surface inspection,Defect detection,Knowledge distillation1.Abstract:本文提出了一种称为类感知对象检测网络(CADN)的弱监督学习方法,仅接受图像标签注释训练,执行图像分类和缺陷定位。通过在分类pipe.
2020-12-11 16:58:30 6795 3
原创 关于numpy矩阵和矩阵索引的数字类型
关于numpy矩阵和矩阵索引的数字类型矩阵整体和矩阵元素1.起因2.测试两种方式的区别3.进一步测试4.结论矩阵整体和矩阵元素1.起因在训练网络时,由于内存原因,把原始的图片归一化代码做了修改,不确定能否解决内存溢出的问题,但是发现了新的问题,其余部分不改变,仅做此修改,网络的训练效果顿时开始原地踏步。images=images/255 #原始代码for i in range(images.shape[0]): images[i]=images[i]/255 #更改后代码
2020-10-22 21:32:47 425
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人