Skip to content

Commit 2d1eb4a

Browse files
authored
Update readme.md
1 parent fdbce99 commit 2d1eb4a

1 file changed

Lines changed: 9 additions & 19 deletions

File tree

readme.md

Lines changed: 9 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,14 @@
1-
## infCNN
2-
A pure numpy-based inference framework for CNN. The infCNN supports the inference of model trained on pytorch.
1+
## numpy-cnn
2+
A pure numpy-based inference framework for CNN. The aim of numpy-cnn is to deploy CNN model with low-cost and few adjustment including embedded systems. Also in order to enlarge the application scope, we support ONNX format, which enables the converting of trained model within various DL frameworks (PyTorch).
33

4-
In general, all the elements used in the inference of CNN are divided into ```op``` and ```layer```. ```op``` contains no trainable weights such as "relu, sigmoid, softmax, maxpool, flatten", ```layer``` contains trainable weights such as "conv2d, dense". The ```op``` and ```layer``` implemented are very few now.
4+
## Features
5+
We establish a extremely streamlined structure to build the CNN inference framework. All the elements (layers, operations, activation fuctions) are abstracted to be ```layer```, and a json formatted ```flow``` is applied to build the computation graph. And the weights are stored in ravel numpy, which largely simplifies the loading process.
56

6-
#### Inference
7-
8-
Example is shown in ```net.py```, which shows the inference of a LeNet CNN. Weights converted from pytorch model are loaded here. To be noted that ```net.py``` should align with ```train/lenet.py``` since they should have the same model. Just replacing ```nn.Conv2d/Linear``` with ```inferCNN.Conv2d/Dense``` will work.
9-
10-
#### Training on pytorch
11-
12-
The training on pytorch is regular, which can be seen in ```train/``` folder. The weights of CNN are exported to ```.mat``` file.
13-
14-
15-
#### Plugins for ImagePy
16-
The plugin example for mnist is in ```plgs/``` folder
17-
18-
#### References
19-
20-
* https://github.com/wiseodd/hipsternet
21-
* https://github.com/pytorch/examples/blob/master/mnist/main.py
7+
## Demos
8+
We have released some demos, which can be investigated inside ```demo/``` folder.
9+
![](https://raw.githubusercontent.com/Image-Py/numpy-cnn/master/demo/craft_text_detector/rst.png)
10+
![](https://raw.githubusercontent.com/Image-Py/numpy-cnn/master/demo/hed_edge_detector/rst.png)
11+
![](https://raw.githubusercontent.com/Image-Py/numpy-cnn/master/demo/resnet18/rst.png)
2212

2313

2414

0 commit comments

Comments
 (0)