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中的所有convblock支持:

版权所有

@Lucas Jin版权所有。在

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
带键的Selenium上的JAVA增量输入数   Java代码已编写,但运行后没有预期的输出   java如何存储文件数据行,并通过它的方法计算成字母,并在另一个文件中打印字母等级?   进行Web服务调用时eclipse中的java ClassNotFoundException   g.drawString()上的graphics2d Java“包装文本”+“自动字体大小”   任务不可序列化Java 1.8和Spark 2.1.1   使用流中现有servlet的JavaMuleESB   Java嵌套数组到递归方法的算法   java为垂直组添加背景   java有没有办法将类似CSV的文本字符串复制到2D数组?   java Android是否使用现有的XML布局将项目动态添加到ListView?   java JPA试图强迫我拥有一个名为“id”的属性和数据库列   java如何基于特定的数据集创建部分数据库转储?   java无法在暂停后恢复音频曲目   java在对象中存储字节数组,然后将其转换为ObjectOutputStream?(卡住了)   java获取SyndEntry的XML字符串值   java IText API将HTML文件转换为PDF格式和图像渲染问题   java我从我的oracle servlet中得到一个空白屏幕