分类:PyTorch
2020
10-08
二分类或分类问题,网络输出为二维矩阵:批次x几分类,最大的为当前分类,标签为one-hot型的二维矩阵:批次x几分类计算百分比有numpy和pytorch两种实现方案实现,都是根据索引计算百分比,以下为具体二分类实现过程。pytorchout=torch.Tensor([[0,3],[2,3],[1,0],[3,4]])cond=torch.Tensor([[1,0],[0,1],[1,0],[1,0]])persent=torch.mean(torch.eq(torch.argmax(out,dim=1),torch.argma...
继续阅读 >
最近做显著星检测用到了NLL损失函数对于NLL函数,需要自己计算log和softmax的概率值,然后从才能作为输入输入[batch_size,channel,h,w]目标[batch_size,h,w]输入的目标矩阵,每个像素必须是类型.举个例子。第一个像素是0,代表着类别属于输入的第1个通道;第二个像素是0,代表着类别属于输入的第0个通道,以此类推。x=Variable(torch.Tensor([[[1,2,1],[2,2,1],[0,1,1]],[[0,1,3],[...
继续阅读 >
先看Pytorch中的卷积classtorch.nn.Conv2d(in_channels,out_channels,kernel_size,stride=1,padding=0,dilation=1,groups=1,bias=True)二维卷积层,输入的尺度是(N,C_in,H,W),输出尺度(N,C_out,H_out,W_out)的计算方式这里比较奇怪的是这个卷积层居然没有定义inputshape,输入尺寸明明是:(N,C_in,H,W),但是定义中却只需要输入in_channel的size,就能完成卷积,那是不是说这样任意size的image都可以进行卷积呢...
继续阅读 >
2020
10-08
1.torch.eq(input,other,out=None)说明:比较元素是否相等,第二个参数可以是一个数,或者是第一个参数同类型形状的张量参数:input(Tensor)----待比较张量other(Tenosrorfloat)----比较张量或者数out(Tensor,可选的)----输出张量返回值:一个torch.ByteTensor张量,包含了每个位置的比较结果(相等为1,不等为0)>>>a=torch.Tensor([[1,2],[3,4]])>>>b=torch.Tensor([[1,1],[4,4]])>>>torch.eq(a,b)tens...
继续阅读 >
1.字面理解:torch.cat是将两个张量(tensor)拼接在一起,cat是concatnate的意思,即拼接,联系在一起。2.例子理解>>>importtorch>>>A=torch.ones(2,3)#2x3的张量(矩阵)>>>Atensor([[1.,1.,1.],[1.,1.,1.]])>>>B=2*torch.ones(4,3)#4x3的张量(矩阵)>>>Btensor([[2.,2.,2.],[2.,2.,2.],[2.,2.,2.],[2.,2.,2.]])>>>C=torch.cat((A,B),0)#按...
继续阅读 >
我就废话不多说了,大家还是直接看代码吧~importtorchinput_tensor=torch.tensor([1,2,3,4,5])print(input_tensor>3)mask=(input_tensor>3).nonzero()print(mask)print(input_tensor.index_select(0,mask))tensor([0,0,0,1,1],dtype=torch.uint8)tensor([3,4])tensor([4,5])补充知识:pytorchtensor筛选满足条件的行或列(使用与或)我就废话不多说了,大家还是直接看代码吧~importtorchx=torch.linspace(1,8,st...
继续阅读 >
max找出tensor的行或者列最大的值:找出每行的最大值:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,1))输出:(tensor([1.,2.,3.]),tensor([0,0,0]))找出每列的最大值:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])print(torch.max(outputs.data,0))输出结果:(tensor([3.]),tensor([2]))Tensor比较eq相等:importtorchoutputs=torch.FloatTensor([[1],[2],[3]])targe...
继续阅读 >