![](https://img-blog.csdnimg.cn/006d996c98744178850605e6b8c5a12c.jpeg?x-oss-process=image/resize,m_fixed,h_224,w_224)
python-pytorch
文章平均质量分 65
python-pytorch代码相关内容记录,主要记录相关库函数使用、torch相关库函数使用等,也记录相关环境问题、运用问题等状况。
tangjunjun-owen
从事人工智能行业,已从事工业检测行业、智慧交通行业、智慧城市行业,已有如下经验:
ARM/jetson部署、NVIDIA性能优化、道路感知2D、单目3D、点云3D、目标跟踪、lidar/radar/vison多传感融合、中间件ROS2开发、相机标定、工业2D检测/分类/分割、数据增强、attention、transformer、模型压缩加速
计算机技能:C++、Python、cuda、tensorrt、ros2
展开
-
transformer图像切块与还原(window_partition+window_unpartition)
假如b ,h,w,c=(3,32,32,768)需将h w按照14尺寸切割,32/14无法整除,需pad为(3,42,42,768)完成固定尺寸块切割,进而完成transformer结构,最终摒弃pad数据还原为(3,32,32,768)。在使用Transformer结构提取特征时,通常会使用window_partition和window_unpartition来划分和还原图像块的过程。这两个步骤是为了将图像分割成小块,送入Transformer网络进行处理,然后再将处理后的特征重新组合成原始图像的尺寸。原创 2024-02-18 18:40:08 · 648 阅读 · 0 评论 -
linux系统服务器A读取远程linux系统服务器B文件(以cv读图文件为列)
当一个服务器需要处理另一个服务器上已知图像的绝对路径时,可以使用 Paramiko 库连接到远程服务器并读取图像,然后使用 OpenCV(cv2)库在本地服务器上加载和处理图像。以下是示例代码,演示了如何使用 Paramiko 和 cv2 库来实现这一操作。本文就给出应用。原创 2024-01-16 21:47:56 · 423 阅读 · 0 评论 -
torch.nn.functional.interpolate与torchvision.transforms.Resize方法对张量图像Resize应用
要使用 PyTorch 对张量进行调整大小,您可以使用 torch.nn.functional.interpolate 函数。要对cpu中类似PIL数据,您可以使用torchvision.transforms.Resize函数。本文将介绍这2种函数应用方法。原创 2024-01-09 11:07:13 · 1227 阅读 · 0 评论 -
环境安装问题(库、代码等问题)
该文章会一直更新遇到环境安装问题,使用他/她人博客解决方法,并附解决方法博客链接。原创 2023-12-20 21:58:43 · 745 阅读 · 0 评论 -
torch.distributed.init_process_group()详细说明(RANK/WORLD_SIZE)
最近在研究deepspeed的方法,得知deepspeed方法也是对pytorch分布式调用的进一步封装,我将回顾以往知识(重温(已忘记了)),之前多数直接使用torch函数调用,或已不太记得之前研究过的东西了,今日特意回顾torch的init_process_group函数,但介于有rank world_size等配置,我特意写此文章详细整理该知识。backend,rank=-1,**kwargs参数说明:backend:指定分布式后端的名称,例如 ‘nccl’、‘gloo’ 或 ‘mpi’。原创 2023-12-13 14:24:36 · 6309 阅读 · 2 评论 -
@classmethod方法介绍
classmethod是用来指定一个类的方法为类方法,没有此参数指定的类的方法为实例方法,说人话无需类实列化调用类中函数或方法的关键字。原创 2023-12-08 11:30:10 · 114 阅读 · 0 评论 -
知识蒸馏KL-loss解读
本文简单介绍知识蒸馏教师模型与学生模型使用KL loss方法。原创 2023-12-01 15:28:38 · 1338 阅读 · 2 评论 -
pythorch的numel()函数计算模型大小与现存占用
本文解释简单给一个模型列子记录如何计算该模型参数量与模型显存占用情况,该文直接调用torchvision库的模型文件构建模型model,在使用parameters()函数遍历,并在遍历情况下使用numel()函数记录模型参数量与显存占用。原创 2023-11-19 21:43:12 · 304 阅读 · 0 评论 -
F.binary_cross_entropy、nn.BCELoss、nn.BCEWithLogitsLoss与F.kl_div函数详细解读
最近我在构建蒸馏相关模型,我重温了一下交叉熵相关内容,也使用pytorch相关函数接口调用,我将对F.binary_cross_entropy()、nn.BCELoss()与nn.BCEWithLogitsLoss()函数做一个说明,同时也简单介绍相对熵的蒸馏F.kl_div()函数做一个介绍。原创 2023-11-15 21:30:14 · 2370 阅读 · 0 评论 -
self.register_buffer方法使用解析(pytorch)
self.register_buffer就是pytorch框架用来保存不更新参数的方法。原创 2023-11-06 22:07:29 · 967 阅读 · 0 评论 -
nn.embedding函数详解(pytorch)
最近发现prompt工程(如sam模型),也有transform的detr模型等都使用了nn.Embedding函数,对points、boxes或learn query进行编码或解码。因此,我想写一篇文章作为记录,本想简单对其 介绍,但写着写着就想把所有与它相关东西作为记录。本文章探讨了nn.Embedding参数、使用方法、模型训练与预测的变化,并附有列子源码作为支撑 ,呈现一个较为完善的理解内容。原创 2023-11-06 21:54:39 · 4202 阅读 · 0 评论 -
路径某个位置更换名称-python
使用python更换某个路径名称,先将其变成路径,在获取路径列表,然后更改你想更换列表对应名称,随后将列表串成路径。原创 2023-09-20 14:41:07 · 101 阅读 · 0 评论 -
np.bincount()极简易懂说明
该函数用于统计输入数组内每个数值出现的次数,输出数组中的索引值对应的是输入数组中的元素值,若输入数组中的某个数值出现了一次,则输出数组对应索引值上的数加1。原创 2023-08-16 15:17:37 · 809 阅读 · 0 评论