setup.py 790 B

1234567891011121314151617181920
  1. from setuptools import setup, find_packages
  2. setup(
  3. name="watermark_verify",
  4. version="1.0",
  5. description="AI模型模型水印提取验证工具",
  6. # 你要安装的包,通过 setuptools.find_packages 找到当前目录下有哪些包
  7. packages=find_packages(),
  8. # 表明当前模块依赖哪些包,若环境中没有,则会从pypi中下载安装
  9. install_requires=['onnxruntime', 'opencv-python~=4.9.0.80', 'numpy']
  10. # install_requires 在安装模块时会自动安装依赖包
  11. # 而 extras_require 不会,这里仅表示该模块会依赖这些包
  12. # 但是这些包通常不会使用到,只有当你深度使用模块时,才会用到,这里需要你手动安装
  13. # extras_require={
  14. # 'pytorch': ['pytorch>=2.0.0'],
  15. # }
  16. )