链式自/至nnoir转换器
nnoir-chainer的Python项目详细描述
nnoir链
从/到nnoir转换器的chainer模型
安装
pip install nnoir-chainer
示例
进口nnoir
import chainer
from nnoir_chainer import NNOIRFunction
m = NNOIRFunction('nnoir_file_path')
x = chainer.Variable(np_array)
with chainer.using_config('train', False):
y = m(x)
print(y)
出口nnoir
m = model.CNN()
chainer.serializers.load_npz('cnn.model', L.Classifier(m))
with chainer.using_config('train', False):
x = chainer.Variable(np.zeros((1, 28*28)).astype(np.float32))
y = m(x)
g = nnoir_chainer.Graph(m, (x,), (y,))
result = g.to_nnoir()
with open('model.nnoir', 'w') as f:
f.buffer.write(result)
这些层由nnoir chainer exporter支持。
- 链节
- 批量规范化
- 偏差
- 线性
- 卷积2d(深卷积2d,展开卷积2d)
- 比例
- 链函数
- 添加
- 添加常量
- 平均池2d
- concat
- 辍学
- 洗脱液
- 漏洞百出
- MaxPooling2d
- mul
- 多常数
- 键盘
- relu
- 重塑
- 乙状结肠
- softmax
- 谭
- 转置
- 未冷却2d