Chainer from/to NNOIR Converter
Project description
nnoir-chainer
Chainer Model from/to NNOIR converter
ATTENTION: This library goes into the maintenance phase too.
Install
pip install nnoir-chainer
Example
Import 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)
Export 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)
These layers are supported by nnoir-chainer exporter.
- chainer.links
- BatchNormalization
- Bias
- Linear
- Convolution2D (DepthwiseConvolution2D, DilatedConvolution2D)
- Scale
- Swish
- chainer.function
- Add
- AddConstant
- AveragePooling2D
- ClippedReLU
- Concat
- Dropout
- ELU
- LeakyReLU
- MaxPooling2D
- Mul
- MulConstant
- Pad
- ReLU
- Reshape
- Sigmoid
- Softmax
- Sub
- Sum
- Tanh
- Transpose
- Unpooling2D
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
nnoir-chainer-1.1.3.tar.gz
(17.2 kB
view hashes)
Built Distributions
nnoir_chainer-1.1.3-py3.7.egg
(145.2 kB
view hashes)
Close
Hashes for nnoir_chainer-1.1.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8191471dc6fc8af0a9edd9789dc85e98e857d49743d4fd695b7f5dd11fab39f2 |
|
MD5 | 08ffe184f2993a89aa1c1add30653857 |
|
BLAKE2b-256 | 2dc80ba3bd361bc45e832c21d0ed8b03b2bce8cff1bcc28d96ced1ded35dd655 |