- 博客(12)
- 收藏
- 关注
原创 torch.profiler分析代码性能
如果分析对象是一个函数,也可以将上述语句放入装饰器中,应用起来更加方便。接下来使用下列语句,'your code'处放置自己的代码。可以看到代码每一环节所消耗的cpu/gpu时间和所占的比例。随后运行程序,就会在控制台获得输出,如下所示。然后在目标函数处应用装饰器即可。
2025-02-05 21:46:20
238
原创 【stable diffusion webui】No module ‘xformers‘. Proceeding without it.
stable diffusion webui运行出现 no module xformers;只需在命令行加上 --xformers即可解决,生成速度提升一倍多!
2024-09-25 13:23:29
660
原创 cv2.imwrite无报错但也没有保存图片
opencv保存图片格式为ndarray,[H, W, C],保存前一定要检查图片格式,否则无法保存,也不会报错,没有任何提示。
2024-08-26 00:02:40
524
原创 SAM本地部署,并从本地加载模型
但是这个代码还有一个问题,就是下面这两行代码需要访问huggingface,但是在批量处理图片的时候,由于网络不稳定,可能会出现失败的情况。(虽然模型会缓存到本地,只用下载一次,但是访问huggingface的流程还是要走的,就可能会出问题)代码前半部分定义了各种功能函数,后半部分是主体代码,其中save_masks函数是我自己写的,增加一个保存mask的功能,其他函数都来自于官方教程。为保证运行稳定,改为从本地加载模型,方法如下(非常简单,但是目前好像没有人说清楚,于是我决定自己记录一下)
2024-06-25 16:53:07
1198
1
原创 Win11开启远程连接后本地复制粘贴失效
本地是win11系统,使用windows远程连接连另一台电脑,连上之后本地复制粘贴功能就失效了。即复制后win+v查看剪贴板,里面没有任何内容,也无法粘贴。有时候使用ctrl+v粘贴后,虽然粘贴不成功,但会导致电脑死机十多秒。好像是todesk共享剪贴板的问题,关闭共享剪贴板后即恢复正常。但是再次开启后也没有碰到问题。如果不行也可以重启todesk试试。
2024-02-21 14:06:35
7446
3
原创 EfficientNet图像分类原理简介
其主要引入了SE通道注意力机制和跳跃连接。SE模块通过显式建模信道之间的相互依赖性来自适应地重新校准信道特性响应,通过该机制,网络可以学习使用全局信息来选择性地强调信息特征并抑制不太有用的特征。而跳跃连接可以有效解决网络训练过程中可能出现的梯度消失问题,有利于构建更深的网络。随后将图像特征展开成一维向量,再经过全连接层和softmax之后即可得到输出向量,即可得到输入图片属于每个类别的置信度。EfficientNet基础网络模型结构如下,除了常规的卷积、池化、全连接之外,其最重要的模块就是MBConv。
2023-06-05 15:12:40
608
1
原创 miniconda3安装后没有anaconda prompt解决方法
第一步:win+R 输入cmd 进入命令行,进入到Anaconda的安装目录,语句:cd/d Anaconda的安装目录。然后在windows搜索栏搜索prompt就可以看到了。
2023-05-17 13:58:00
3373
3
原创 模型预测过程中报错 RuntimeError: Sizes of tensors must match except in dimension 1.
调用YOLOv7检测时报错:RuntimeError: Sizes of tensors must match except in dimension 1. Expected size 58 but got size 57 for tensor number 1 in the list.
2022-12-05 15:34:10
6476
12
原创 模型预测过程中:cuda out of memory
yolov7预测过程中cuda out of memory,加了torch.cuda.empty_cache()依然无效
2022-11-29 22:58:51
1334
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人