nne

Install

python -m pip install -e.

Example

onnx

importnneimporttorchvisionimporttorchimportnumpyasnpinput_shape= (1,3,64,64)onnx_file='resnet.onnx'model=torchvision.models.resnet34(pretrained=True).cuda()nne.cv2onnx(model,input_shape,onnx_file)

tflite

importtorchvisionimporttorchimportnumpyasnpimportnneinput_shape= (10,3,224,224)model=torchvision.models.mobilenet_v2(pretrained=True).cuda()tflite_file='mobilenet.tflite'nne.cv2tflite(model,input_shape,tflite_file)

tflite(edgetpu)

importtorchvisionimporttorchimportnumpyasnpimportnneinput_shape= (10,3,112,112)model=torchvision.models.mobilenet_v2(pretrained=True)tflite_file='mobilenet.tflite'nne.cv2tflite(model,input_shape,tflite_file,edgetpu=True)

TensorRT

importnneimporttorchvisionimporttorchimportnumpyasnpinput_shape= (1,3,224,224)trt_file='alexnet_trt.pth'model=torchvision.models.alexnet(pretrained=True).cuda()nne.cv2trt(model,input_shape,trt_file)

Script

$nne -husage: nne [-h] [-a ANALYZE_PATH] [-s SIMPLYFY_PATH] [-t TFLITE_PATH] model_pathNeural Network Graph Analyzerpositional arguments:  model_path            model pathfor analyzingoptional arguments:  -h, --help            show thishelp message andexit  -a ANALYZE_PATH, --analyze_path ANALYZE_PATH                        Specify the path to output the Node information of the modelin json format.  -s SIMPLYFY_PATH, --simplyfy_path SIMPLYFY_PATH                        onnx model to simplyfier  -t TFLITE_PATH, --tflite_path TFLITE_PATH                        onnx model to tflite

Support Format

License