
pytorch
文章平均质量分 55
prinTao
我是prinTao,计算机视觉与机器人方向。欢迎关注我,一起学习CV~
展开
-
【论文阅读】DynamicControl :一种新的controlnet多条件控制方法
现有方法要么处理条件效率低下,要么使用固定数量的条件,这并不能完全解决多个条件的复杂性及其潜在冲突。这强调了需要创新方法来有效管理多种条件,以实现更可靠和详细的图像合成。为了解决这个问题,我们提出了一个新的框架 DynamicControl ,它支持不同控制信号的动态组合,允许自适应选择不同数量和类型的条件。本文方法从一个双循环控制器开始,它通过利用预先训练的条件生成模型和判别模型为所有输入条件生成初始真实分数排序。此控制器评估提取条件和输入条件之间的相似性,以及与源图像的像素级相似性。原创 2024-12-25 15:26:08 · 1498 阅读 · 0 评论 -
torch.fft 出现 ComplexHalf 或 Half 不支持
【代码】torch.fft 出现 ComplexHalf 或 Half 不支持。原创 2024-10-16 14:53:27 · 808 阅读 · 0 评论 -
【论文+代码】VISION PERMUTATOR 即插即用的多层感知器(MLP)模块
函数是模型的主要执行路径,它将输入图像转化为嵌入特征,通过网络模块进行处理,归一化后通过分类头部输出预测结果。这一步步的处理确保了模型能够有效地提取图像特征,并进行准确的分类。代码链接 https://github.com/houqb/VisionPermutator/blob/main。本文的研究成果在项目的实现过程中起到了至关重要的作用。核心算法和模型训练。原创 2024-06-06 10:49:40 · 622 阅读 · 0 评论 -
使用Docker配置深度学习环境——以diffusers为例
Docker的其他信息可以在我的网站上找到,这里假设安装完成了,直接上手。把放到最前面。原创 2024-05-15 15:43:55 · 399 阅读 · 0 评论 -
如何将数据(通常是张量)和模型转移到CUDA设备(即GPU)上
在PyTorch中,将模型和数据移到GPU上进行加速计算是深度学习训练过程中的一个常见步骤。这里将介绍如何将数据(通常是张量)和模型转移到CUDA设备(即GPU)上。原创 2024-04-11 20:02:56 · 4648 阅读 · 0 评论 -
【小工具】一键美化nvidia-smi,查看GPU运行状态,无须配置
可以在远程console看,是动态的。鼠标滚轮控制显示哪个进程。原创 2024-03-14 10:56:06 · 571 阅读 · 0 评论 -
【CUDA】获取当前显存状态的多种方式:准确性与原理讨论
显存状态的获取对于深度学习任务至关重要,特别是在使用 GPU 进行模型训练和推理时。了解当前显存的使用情况可以帮助我们优化算法、调整超参数,以及有效地管理计算资源。在本文中,我们将讨论多种获取当前显存状态的方法,并探讨它们的准确性和原理。原创 2024-03-12 09:46:34 · 4904 阅读 · 1 评论 -
【debug】Pytorch环境冲突
创建一个新的conda环境:如果在当前环境中解决冲突比较困难,考虑创建一个新的conda环境,并在这个新环境中安装需要的包,可以有效避免版本冲突。统一环境中的PyTorch版本:确保pytorch、torchvision和torchaudio都是兼容同一版本的PyTorch。因为torchaudio和torchvision指定了需要特定版本的pytorch,而cudatoolkit又指定了需要特定版本的vs2015_runtime。1.10.1,所以在这一点上它们是兼容的。原创 2024-02-22 16:07:29 · 1371 阅读 · 1 评论 -
【cuda】RuntimeError: Unexpected error from cudaGetDeviceCount()
会发现 nvidia-smi 不管用了。不要立即重启,会黑屏的。赶紧记录使用的驱动版本号,最好找到安装包位置。直接重装原版驱动,环境还能用。参考我的安装博客。已经黑屏就进入安全模式,network模式。卸载可视化桌面和显卡驱动。驱动程序已经更新,可以尝试回滚到之前的版本。原创 2024-01-25 20:52:36 · 1235 阅读 · 0 评论 -
【一次性解决】深入理解CUDA和PyTorch的安装与多版本管理的三种方式
系统环境变量很多人配置环境就是直接安装三件套,而对于版本管理不是很清楚。在开发初期,这样做没什么问题。但是如果服务器多人使用,或者复现代码多(pytorch版本和cuda版本是互相依赖的),就需要更进一步的版本管理方法。这里将详细介绍用于深度学习或者CPP开发的CUDA版本管理方式。在配置环境前,首先需要明确自己的需求。对于使用Python进行的开发,推荐在。对于C++开发,只使用手动管理版本即可,无须安装anaconda和docker。请注意,以下的方法选择其一即可,混着用容易把自己搞乱了。原创 2024-01-25 20:09:07 · 2236 阅读 · 0 评论 -
【TSP数据集】用于 加载ATSP和TSP 测试样例的python加载器【使用所有类型数据】【使用数据给出方法计算邻接矩阵和邻接表】
class TSP_DATA: '''用于读取数据、存放图的数据结构 类内变量: 必有: self.NAME 测试样例名称 self.TYPE 测试样例类型 TSP ATSP self.DIMENSION. 维度 self.EDGE_WEIGHT_TYPE 边权值计算方式 决定读取方式 self.matrix 矩阵形式数据 self.table .原创 2022-05-30 22:12:53 · 1411 阅读 · 0 评论 -
【grid】pytorch中的Flow_filed,MES,affine_gridHGRID,GRID_SAMPLE详解
specifiesinputxandy根据输入值和映射网格(flow-field grid)计算输出。它主要用于在图像处理和计算机视觉任务中,根据给定的网格对输入数据进行采样和插值。提供一个input的Tensor以及一个对应的flow-field网格(比如光流,体素流等),然后根据grid中每个位置提供的坐标信息(这里指input中pixel的坐标),将input中对应位置的像素值填充到grid指定的位置,得到最终的输出。原创 2024-01-14 18:10:22 · 1859 阅读 · 0 评论 -
迈向更高质量的深度估计
目前的深度估计从业人员大多follow the settings of specific backbone without thinking about why is that。本文将详细探索从数据(不同类型数据集加载、稳定性、预处理、数据生成),超参数设置(学习率、损失函数)到训练过程的不同技巧。通过在您的pipeline中检查这些设定,可能有助于提升自己训练模型,或是将深度估计用于downstream task的性能。原创 2024-01-04 10:58:14 · 972 阅读 · 0 评论 -
【pytorch】自动求导机制
forward() 是执行该操作的代码。它可以需要 你想要多少个参数,其中一些是可选的,如果你 指定默认值。这里接受所有类型的 Python 对象。Tensor 跟踪历史记录的参数(即, requires_grad=True)将被转换为不跟踪历史记录的内容 在调用之前,它们的使用将被记录在图表中。请注意,这 逻辑不会遍历列表/字典/任何其他数据结构,只会 考虑作为调用的直接参数的张量。你可以 返回单个 Tensor 输出,或 tuple 张量(如果有多个输出)。原创 2023-12-20 16:20:29 · 1573 阅读 · 0 评论 -
【自动驾驶】鸽了很久的小物体目标检测代码【小物体目标检测】
鸽了很久的小物体目标检测代码https://github.com/LT1st/SmallObstacleDetection/tree/main/code#readmePytorch: Small ObstacleTODO Modify for 4 channel inputTrainingFollow below steps below to train your model:Input arguments: (see full input arguments via python t原创 2022-04-26 21:42:17 · 1583 阅读 · 1 评论 -
【代码积累】机器学习常用
工具from tqdm import tqdmtqdm.pandas()a = tqdm([for i in a])数据处理数据集库import osimport sys文件IO多用withwith open(f'{HOME_DIR}/YoloR-data.yaml', 'w') as outfile:路径管理库拼接os.path.join自己写def add_path(row): return f"{COTS_DATASET_PATH}/video_{row原创 2022-01-10 09:23:55 · 165 阅读 · 0 评论 -
【pytorch】安装好以后jupyter的配置
拿到了新机器,半天时间配置了环境,发现了以前没关注到的点。从conda环境启动的jupyter不一定会是对应安装torch的那个。conda activate pytorch_gpuconda install ipykernelpython -m ipykernel install --user --name pytorch_gpu保证安装好了kernel的驱动...原创 2022-01-06 19:32:13 · 1004 阅读 · 0 评论 -
【minitorch】Day2:Task0和测试模块
单元测试单元测试应该是快速简单的代码片段,可以在开发时快速运行。一个聪明的中间立场是使用随机属性检查。这种方法由名为 QuickCheck 的库。随机属性检查的另一个好处是它实际上使测试更短、更容易编写,因为它为您生成案例。在 MiniTorch 中,我们将使用 Python 中一个很棒的属性检查库,称为Hypothesis ( https://hypothesis.readthedocs.io/ )。假设预定义了一整套构建块策略,用户可以在编写测试时从中选择。(你也可以编写你的策略,你将在下一个作业中完原创 2021-12-30 22:54:20 · 240 阅读 · 0 评论 -
【argparse】【python】
argparseIntro.sample特殊Intro.argsparse是python的命令行解析的标准模块,内置于python,不需要安装。这个库可以让我们直接在命令行中就可以向程序中传入参数并让程序运行。sampleimport argparseparser = argparse.ArgumentParser(description='命令行中传入一个数字')#type是要传入的参数的数据类型 help是该参数的提示信息parser.add_argument('integers',原创 2021-12-14 21:33:08 · 477 阅读 · 0 评论 -
[pytorch][报错] RuntimeError: Can‘t call numpy() on Tensor that requires grad.detach().numpy() instead
tensor转换numpy失败报错原因解决报错原因RuntimeError: Can’t call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.出现问题的地方:解决使用Tensor.detach().numpy()Tensor.detach().numpy()由于是跑开源代码遇到的问题,作者写的也不好看,只好直接改tensor。不推荐这么做,但是作者传参不注意,也没得办法了。...原创 2021-12-14 19:17:32 · 7254 阅读 · 2 评论