![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
源码解析
zhuikefeng
人工智能爱好者
展开
-
BeamSearch算法原理及代码解析
1.算法原理beam search有一个超参数beam_size,设为 k 。第一个时间步长,选取当前条件概率最大的 k 个词,当做候选输出序列的第一个词。之后的每个时间步长,基于上个步长的输出序列,挑选出所有组合中条件概率最大的 k 个,作为该时间步长下的候选输出序列。始终保持 k 个候选。最后从k 个候选中挑出最优的。2.中心思想假设有n句话,每句话的长度为T。encoder的输出shape为(n, T, hidden_dim),扩展成(n*beam_size, T, hidden_dim)原创 2022-04-07 19:52:51 · 3069 阅读 · 1 评论 -
MMdetection的Proposal原理和代码解析
一、算法原理接受N级score,bbox_pred,anchor和image_shape作为输入,通过anchor和框的偏移(bbox_pred)得到proposal,然后对这些proposal做NMS,最后选出前num个。二、执行步骤将每级score,bbox_pred,anchor按照score从大到小排序,并选择前num_pre个(一般为1000),共N*num_pre个。 通过anchor和框的偏移(bbox_pred)得到proposal 去除框大小为负数的框,并且对于每级的pro原创 2022-02-25 16:38:17 · 3563 阅读 · 0 评论 -
RoiAlign源码解析
/** 参数解释* bottom_data:输入的特征图数据,shape是(n,c,h,w)* bottom_rois:输入的roi数据,shape是(num_rois,5),第一维表示共有多少个rois,第二维是[batch_index,x1,y1,x2,y2],第一个值表示rois所在的batch_id,后四个是rois所在原图的坐标值,可以根据spatial_scale对应到特征图上* nthreads:等于roipooling后输出的size,即num_rois*channels*pool.原创 2021-12-14 17:44:40 · 2487 阅读 · 0 评论 -
NMS源码解析
一、IOU的概念二、NMS的算法原理 选取该类box中scores最大的一个,记为box_best,并保留它 计算box_best与其余的box的IOU 如果其IOU>threshold了,就舍弃这个box(因为可能这两个box表示同一目标,保留分数高的哪一个) 从最后剩余的boxes中,再找出最大scores的哪一个,如此循环往复,直到没有box为止 三、源码解析# ---------------------------# 非极大值抑制(N..原创 2021-12-14 20:55:36 · 4453 阅读 · 0 评论 -
onnxruntime编译选项解析
enable_nvtx_profile:在ORT使用NVTX profile。NVTX(NVIDIA® Tools Extension SDK)是一个提供事件注解,编码ranges和资源的c接口程序库use_horovod:使用horovod。Horovod是Uber开源的基于Ring-AllReduce方法的深度学习分布式训练工具,以支持多种流行架构包括TensorFlow、Keras、PyTorch等msvc_toolset:使用msvc工具集。msvc是微软的编译器vcu...原创 2020-09-18 19:58:54 · 4097 阅读 · 0 评论 -
myTinySTL源码解析:vector
一、测试程序#ifndef MYTINYSTL_VECTOR_TEST_H_#define MYTINYSTL_VECTOR_TEST_H_// vector test : 测试 vector 的接口与 push_back 的性能#include <vector>#include "../MyTinySTL/vector.h"#include "test.h"namespace mystl{namespace test{namespace vector_test原创 2020-06-28 19:37:58 · 1839 阅读 · 2 评论 -
cJSON源码解析(超级详细!!!)
1.cJSON结构体定义typedef struct cJSON { struct cJSON *next,*prev; /* next/prev allow you to walk array/object chains. Alternatively, use GetArraySize/GetArrayItem/GetObjectItem */ struct cJSON *child; /* An array or object item will have a child pointer p原创 2020-06-26 21:35:06 · 4413 阅读 · 1 评论