![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
哈宝可爱
这个作者很懒,什么都没留下…
展开
-
MobileNetV3代码和网络
MobileNetV3实现参考博客:https://zhuanlan.zhihu.com/p/323346888https://blog.csdn.net/qq_43739296/article/details/122338041来自paddelocr中的MobileNetV3,实现网络的类如下。ConvBNLayerConvBNLayer类实际上就是实现了一个卷积层+bn层,如名所示。当我们写其他模块需要用到各种卷积层如1x1卷积和depth-wise卷积时,就可以通过传入各种相关参数给它实原创 2022-01-06 14:59:07 · 1859 阅读 · 0 评论 -
配置full_auto_read_meter pytorch 环境
配置full_auto_read_meter pytorch 环境所需pytorch环境需要torch torchvision和paddle,由于调用paddleocr,paddleocr需要paddle2.0和cuda10.0或10.1或10.2环境安装我的电脑只有cuda10.0,首先按照官网的指示安装:conda install pytorch torchvision发现自动安装cuda11,吓死人了,赶紧按n,重新开了一个环境:conda create -n full_auto_rea原创 2022-01-06 10:41:22 · 202 阅读 · 0 评论 -
nn.CrossEntropyLoss()的计算过程
nn.CrossEntropyLoss()中的计算方法:原公式为:loss = -(y1logp1 + y2logp2 + y3logp3 + … + ynlogpn)y1, y2, …,yn为ground-truth标签,p1, p2, …, pn为prediction计算时,y3=1, 其他类别的标签=0, 原公式等同于:loss = -logp3可以看到:真实标签被处理为one-hot型交叉熵公式的计算已经包含了softmax,因此网络中以及网络得到的结果不再需要进行softmax原创 2021-10-25 15:02:51 · 1353 阅读 · 0 评论 -
【代码复盘】 找出inference结果中检测困难的图片 find_hard_pics.py
文章目录任务代码思路实现的细节:代码层面技术细节:代码任务阈值划到0.3或0.35,把所有图片检测一下,找出那些检测困难的图片,把txt收集起来。检测困难,是指一张图片中,所有目标(或大部分)都检测不出来,这样的图片可能采集来源有点问题,或者有其他问题。我们需要把这样的图片找出来。代码思路如果一张图片的所有目标或大部分目标检测不出来,那么这张图片的Positive中,TP很少,所以我们需要计算出一张图片的TP占gt label的比例,如果比例很小(如小于20%),那么这张图片就属于检测困难的图片原创 2021-09-14 20:16:20 · 204 阅读 · 0 评论 -
pytorch nonzero(), torch.unique()
nonzero()以张量的形式返回非0元素的索引,不管有几维,维度索引都放在一个“列表”中import torcha =torch.Tensor([1,2,4,4,5])print(torch.nonzero(a<4))torch.unique()挑出tensor中的独立不重复元素。原创 2021-09-10 16:17:02 · 511 阅读 · 0 评论 -
2021-09-09 找出检测困难的图片
文章目录任务思路首先要确定我们需要做个什么样的程序,在什么情况下运行,输入输出是什么任务阈值划到0.3或0.35,把所有图片检测一下,找出那些检测困难的图片,把txt收集起来。检测困难,是指一张图片中,所有目标(或大部分)都检测不出来,这样的图片可能采集来源有点问题,或者有其他问题。我们需要把这样的图片找出来。思路首先要确定我们需要做个什么样的程序,在什么情况下运行,输入输出是什么...原创 2021-09-09 17:57:21 · 72 阅读 · 0 评论 -
pytorch max()
文章目录用法用法torch.max(input, dim, keepdim=False, out=None) -> (Tensor, LongTensor)它是一个过载函数,函数参数不同函数的功能和返回值也不同。input是一个tensor.当max函数中有维数参数的时候,它的返回值为两个,一个为最大值,另一个为最大值的索引.当维度为0时,并不是对每行求值,而是对每列求值,返回shape为一行的结果。当维度为1时,并不是对每列求值,而是对每行求值,返回的结果为一列。如:# 对每一行原创 2021-09-09 16:06:12 · 105 阅读 · 0 评论 -
Pytorch-book运行环境准备
Pytorch-book运行环境准备1. 准备工作1.1 代码1.2 环境1.3 配置jupyter notebook1.4 为jupyter notebook选择环境1. 准备工作1.1 代码下载Pytorch-book中文版即可1.2 环境本书是1.0.1,因此建conda环境:# 不要忘了写环境名字conda create -n pytorchbook python=3.6 pytorch=1.0.1 torchvision1.3 配置jupyter notebook内容在jupy原创 2020-11-27 17:49:06 · 108 阅读 · 0 评论