深度学习
HarryXxc
这个作者很懒,什么都没留下…
展开
-
模型的计算量与参数量计算
最近在魔改网络,发现对于模型的计算量和参数量要有一定的概念才行我直接使用的是thop包pip install thopfrom thop import profileimport torch.nnimport torchvisionfrom efficientnet_pytorch import EfficientNetresnet_18 = torchvision.models.resnet18()resnet_34 = torchvision.models.resnet34().原创 2022-03-31 10:53:01 · 593 阅读 · 0 评论 -
对于工业缺陷小样本问题的总结
总结文章原文:如何解决工业缺陷检测小样本问题?工业缺陷样本难以获得,那么需要从 源头(缺陷样本) 或者 算法 这两个角度考虑。从样本上考虑,无非就是增加样本量(1)手动给正常样本制造缺陷,但是这个费时费力,并且对于昂贵的产品不合适。(2)利用仿真,但是现在大多这种数据增广的方式利用的是传统的方案,只能仿真出缺陷的位置改变或者一些简单的纹理变化,对于复杂的纹理以及缺陷的各种形状还比较难。对于以上两种增加样本的理解,我认为一方面是费时费力的,另外一方面因为样本的种类各种各样,没有办法预测,所以.原创 2022-02-25 13:45:31 · 1395 阅读 · 0 评论 -
transforms中RandomResizedCrop、Resize、CenterCrop的理解
首先要记住,transforms只能对PIL读入的图片进行操作,而且PIL和opencv只能读取H * W * C形式的图片transforms.RandomResizedCrop(size) : 将原图片随机裁剪出一块,再缩放成相应 (size*size) 的比例import matplotlib.pyplot as pltfrom PIL import Imagefrom torchvision import transformsfile_path = "./flower.jpg"i.原创 2022-02-09 15:01:06 · 17961 阅读 · 1 评论 -
本地电脑连接实验室服务器使用tensorboard
首先在命令行使用原创 2022-01-25 11:32:28 · 1449 阅读 · 1 评论 -
matlab去除红眼代码及详细介绍
本篇博客借鉴了:https://blog.csdn.net/qq_35608277/article/details/78363139但是他的代码还有一些地方没有说明清楚,并且使用之后效果不是很好。所以我在他的基础上再整理了一下关于去除红眼的模型我这里使用的是HSI模型,HSI转RGB和RGB转HSI的公式大家应该都知晓了,我这里就直接放图了。我们确定好ROI区域后,就根据以下公式来消除红眼:下面直接上代码RGB转HSI代码function [hsi,H,S,I]=rgb2hsi(rg.原创 2021-02-09 12:01:13 · 2908 阅读 · 2 评论 -
pytorch学习-自动求梯度
这段时间刚开始学习pytorch,记录下学习的一些基础知识吧。Pytorch中的Tensor的属性.requires_grad如果设置为True,它将开始追踪在其上的所有操作(这样就可以利用链式法则进行梯度传播)。完成计算后,可以调用.backward()来完成所有梯度计算。此Tensor的梯度将累积到.grad属性中。# 这里需要注意,Only Tensors of floating point and complex dtype can require gradientsx = torch.原创 2021-01-24 19:21:02 · 451 阅读 · 0 评论 -
Anaconda中安装torchtext和torchvision遇到的问题
在跑如下例程代码的时候,会报错:ModuleNotFoundError :No module named ‘torchtext’import torchimport torchvisionimport torchvision.transforms as transformsimport matplotlib.pyplot as pltimport timeimport syssys.path.append("..") # 为了导入上层目录的d2lzh_pytorchimport d2lzh_原创 2020-11-21 19:45:19 · 1820 阅读 · 0 评论