jf_pmFSk4VX的个人主页_电子工程师 - 电子技术论坛

jf_pmFSk4VX

经验：积分：14

查看完整档案 >>

+关注发私信

他的动态

发布了文章 2023-2-1 14:26

PyTorch构建自己一种易用的计算图结构

PNNX项目 PyTorch Neural Network eXchange(PNNX)是PyTorch模型互操作性的开放标准....

0

1948次阅读

0条评论
发布了文章 2023-1-30 09:34

总结FasterTransformer Encoder(BERT)的cuda相关优化技巧

FasterTransformer BERT 包含优化的 BERT 模型、高效的 FasterTransformer 和 INT8 量化推理。...

0

4333次阅读

0条评论
发布了文章 2023-1-13 14:38

Loss计算详细解析

分类损失(cls_loss)：该损失用于判断模型是否能够准确地识别出图像中的对象，并将其分类到正确的类别中。...

0

5261次阅读

0条评论
发布了文章 2023-1-13 11:51

张量类Tensor的实现

在Tensor张量中，共有三维数据进行顺序存放，分别是Channels(维度)，Rows(行高), Cols(行宽)，三维矩阵我们可以看作多个连续的二维矩阵组成，最简单的方法就是使用嵌套的vector数组...

0

1369次阅读

0条评论
发布了文章 2023-1-12 15:05

NIVDIA的reduce优化笔记

通俗的来说，Reduce就是要对一个数组求 sum，min，max，avg 等等。Reduce又被叫作规约，意思就是递归约减，最后获得的输出相比于输入一般维度上会递减。...

0

1222次阅读

0条评论
发布了文章 2023-1-11 17:07

Relu算子的介绍

Relu是一种非线性激活函数，它的特点有运算简单，不会在梯度处出现梯度消失的情况，而且它在一定程度上能够防止深度学习模型在训练中发生的过拟合现象。...

0

2443次阅读

0条评论
发布了文章 2023-1-11 15:40

YOLOv5中的autoanchor.py代码解析

我们不用手工选择 anchor boxes，而是在训练集的边界框上的维度上运行 K-means 聚类算法，自动找到良好的 anchor boxes 。如果我们使用具有欧几里得距离的标准 K-means，那么较大的框会比较小的框产生更多的误...

0

2497次阅读

0条评论
发布了文章 2023-1-10 09:51

Max Pooling算子的设计实现

池化层在深度学习网络中的作用一般是用来缓解卷积层对位置的过度敏感性。...

0

1825次阅读

0条评论
发布了文章 2023-1-8 15:25

OneFlow elementwise模板

众所周知，atomicAdd是CUDA中非常昂贵的操作，特别是对于half类型来说 atomicAdd 巨慢无比，慢到如果一个算法需要用到 atomicAdd，那么相比于用 half ，转成 float ，再 atomicAdd，再转回去还...

0

993次阅读

0条评论
发布了文章 2023-1-8 14:29

分享一种基于深度图像梯度的线特征提取算法download

在低纹理区域，传统的基于特征点的SfM/SLAM/三维重建算法很容易失败。因此很多算法会尝试去提取线特征来提高点特征的鲁棒性，典型操作就是LSD。...

0

2012次阅读

0条评论
发布了文章 2023-1-5 11:29

车道线检测Ultra Fast Deep Lane Detection V2讲解

为了解决上述的定位错误问题，文章便提出一种混合锚点机制，中间水平角度大的车道线使用横向锚点来表达，两侧水平角度小的车道线用纵向锚点来表达。每条车道线都用一组归一化坐标来表示...

0

5920次阅读

0条评论
发布了文章 2023-1-3 17:08

一种基于new concepts的text-to-image生成模型的fine-tuning方法

通过简单文本prompts，用户能够生成前所未有的质量的图像。这样的模型可以生成各种各样的对象、风格和场景，并把它们进行组合排序，这让现有的图像生成模型看上去是无所不能的。...

0

1578次阅读

0条评论
发布了文章 2022-12-30 10:43

YOLOv5解析之downloads.py 代码示例

会调用上面的 safe_download 函数。会用在 experimental.py 中的 attempt_load 函数和 train.py 中，都是用来下载预训练权重。...

0

1131次阅读

0条评论
发布了文章 2022-12-23 15:08

解析OneFlow BatchNorm相关算子实现

可以看到 CUDNN_BATCHNORM_PER_ACTIVATION 被用于非卷积层，在OneFlow中只有当输入Tensor的维度为2时才选取这种模式。而CUDNN_BATCHNORM_SPATIAL_PERSISTENT这种模式只有当...

0

1209次阅读

0条评论
发布了文章 2022-12-19 11:10

3D检测算法的smoke PyTorch 实现

在下图中将小车沿着y轴顺时针旋转，待小车和camera连线与相机坐标系的z轴重合时停止，那么紫色的角是没有发生变化的。...

0

1159次阅读

0条评论