-
nn-common-modules: Pytorch实现的通用模块、块和损失,专门应用于细分模型的CNN
资源介绍
NN通用模块
可以在深层神经网络中专门用于分段的常见模块,块和损失,请使用以获取API手册的参考
该项目有3个模块
损失(losses.py)->它具有定义为python类的所有损失函数
骰子损失
尤洛斯
CrossEntropyLoss2d
合并亏损
焦点损失
模块(modules.py)->它具有FCN的所有常用构建基块
密集块
编码器块
解码器块
分类器块
通用块
SDNetEncoderBlock
SDNetDecoderBlock
八度密集块
OctaveEncoderBlock
OctaveDecoderBlock
贝叶斯模块(bayesian_modules.py)->它具有贝叶斯FCN的所有常用构建基块
贝叶斯卷积块
贝叶斯编码器块
贝叶斯解码器块
贝叶斯分类器块
入门
先决条件
您需要具备以下条件才能使该库按预期工作
Python> = 3.5
火炬> = 1