setup.py 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. #!/usr/bin/env python
  2. import setuptools
  3. setuptools.setup(
  4. name="netron",
  5. version="0.0.0",
  6. description="Viewer for neural network, deep learning, and machine learning models",
  7. long_description='Netron is a viewer for neural network, deep learning, and machine learning models.\n\n' +
  8. 'Netron supports ONNX, TensorFlow Lite, Keras, Caffe, Darknet, ncnn, MNN, PaddlePaddle, Core ML, MXNet, RKNN, MindSpore Lite, TNN, Barracuda, Tengine, TensorFlow.js, Caffe2 and UFF. Netron has experimental support for PyTorch, TensorFlow, TorchScript, OpenVINO, Torch, Vitis AI, Arm NN, BigDL, Chainer, CNTK, Deeplearning4j, MediaPipe, ML.NET and scikit-learn.',
  9. keywords=[
  10. 'onnx', 'keras', 'tensorflow', 'tflite', 'coreml', 'mxnet', 'caffe', 'caffe2', 'torchscript', 'pytorch', 'ncnn', 'mnn', 'openvino', 'darknet', 'paddlepaddle', 'chainer',
  11. 'artificial intelligence', 'machine learning', 'deep learning', 'neural network',
  12. 'visualizer', 'viewer'
  13. ],
  14. license="MIT",
  15. package_dir={ 'netron': 'netron' },
  16. packages=[ 'netron' ],
  17. package_data={ 'netron': [ '*.*' ] },
  18. exclude_package_data={ 'netron': [ 'app.js', 'electron.*' ] },
  19. install_requires=[],
  20. author='Lutz Roeder',
  21. author_email='[email protected]',
  22. url='https://github.com/lutzroeder/netron',
  23. entry_points={ 'console_scripts': [ 'netron = netron:main' ] },
  24. classifiers=[
  25. 'Intended Audience :: Developers',
  26. 'Intended Audience :: Education',
  27. 'Intended Audience :: Science/Research',
  28. 'Programming Language :: Python :: 3',
  29. 'Programming Language :: Python :: 3.6',
  30. 'Topic :: Software Development',
  31. 'Topic :: Software Development :: Libraries',
  32. 'Topic :: Software Development :: Libraries :: Python Modules',
  33. 'Topic :: Scientific/Engineering',
  34. 'Topic :: Scientific/Engineering :: Mathematics',
  35. 'Topic :: Scientific/Engineering :: Artificial Intelligence',
  36. 'Topic :: Scientific/Engineering :: Visualization'
  37. ]
  38. )