nbpython是用于快速网络原型的neral网络生成器
nb-python的Python项目详细描述
NB公司
神经网络块(又名:NB,或神经网络构建器)。这个库为你提供大量的花式积木,快速导入,打造你的强大。一些SOTA技巧和连接,如CSP,ASFF,Attention,BaseConv,Hardswish,所有这些都包括在快速原型你的模型。在
nb是一个来自工程的想法,我们用一些常见的模块建立模型,我们用SOTA技巧探索新的想法,但是所有这些东西都可以集中在一个地方,并用于模型的快速设计和原型设计。在
这个项目目前正在建设中,一旦我发现一些新的块,我将迅速更新它,在模型中真正起作用。另外,每个更新的块都将被记录在更新中。在
使用
下面是一个使用NB构建YoloV5的例子!在
构建conv层的一个简单示例:
^{pr2}$请注意,我们使用cfg
到特定规范和激活的原因是为了让用户以yaml格式动态地切换他们的模型配置,而不是硬编码。在
更新
2020.09.12:添加了新的主干网:
SpineNet是一个专门用于检测的主干模型,它是一个主干,但可以做FPN的事情!!更多信息请参考谷歌的论文link。在
fromnb.torch.bakbones.spinenetimportSpineNetmodel=SpineNet()
在2020.09.11:新增区块:
在resnet.Bottleneck resnet.BasicBlock ConvBase
支持矩阵
我们在此处列出nb中的所有conv
和block
支持:
conv
:- 转换
- 转换:https://arxiv.org/pdf/1903.10520.pdf
- 。。。在
Blocks
:- CSP块:
版权所有
@Lucas Jin版权所有。在
- 项目
标签: