![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
我有一個夢想
IT
展开
-
Focal Loss pytorch实现
引用自知乎 以备后用实现import torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd import Variableclass FocalLoss(nn.Module): r""" This criterion is a implemenation of Focal Loss, which is proposed in Focal Los..原创 2022-05-16 16:43:45 · 621 阅读 · 0 评论 -
Pytorch permute函数
torch.permute()>>> x = torch.randn(2, 3, 5)>>> x.size()torch.Size([2, 3, 5])>>> x.permute(2, 0, 1).size()torch.Size([5, 2, 3])原创 2021-04-28 11:30:08 · 443 阅读 · 0 评论 -
pytorch torch.cat()
按行拼接 或者 按列拼接>>> x = torch.randn(2, 3)>>> xtensor([[ 0.6580, -1.0969, -0.4614], [-0.1034, -0.5790, 0.1497]])>>> torch.cat((x, x, x), 0)tensor([[ 0.6580, -1.0969, -0.4614], [-0.1034, -0.5790, 0.1497], ..原创 2021-04-23 10:36:41 · 279 阅读 · 0 评论 -
ONNX‘s Upsample/Resize operator did not match Pytorch‘s Interpolation until opset 11.
pytorch转onnx时出现的 torch.onnx.export默认使用opset_version=9 改为11就行原创 2021-04-14 11:09:13 · 862 阅读 · 0 评论 -
pytorch ReLU 中的inplace
inplace = True:不创建新的对象,直接对原始对象进行修改; inplace = False:对数据进行修改,创建并返回新的对象承载其修改结果。默认是False,即创建新的对象进行修改,原对象不变,和深复制和浅复制有些类似。...原创 2020-05-27 16:23:16 · 540 阅读 · 0 评论 -
Faster R-CNN模型工程部署(pytorch)
前言因为本人最近在做一个与目标检测有关的项目,最开始只是跑一跑模型,但是后来乙方提出要求需要有我们提出部署方案并实施,最开始在网上查了许多资料,发现现在有许多都是tensorflow的一些部署,而关于pytorch部署这方面实际的资料着实很少,鉴于自己最后调了出来,因而想着写一个博客和大家分享一下,相互学习学习。1.FLaskFlask是一个 web 框架。也就是说 Flask 为你提供工...转载 2019-12-17 09:32:58 · 767 阅读 · 2 评论