setup.py 2.2 KB

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