- 博客(3)
- 收藏
- 关注
RSS订阅原创 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
3993
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
6633
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
306
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人