watermark_generate_pkg/setup.py

from setuptools import setup, find_packages

setup(
    name="watermark_generate",
    version="1.0",
    description="AI模型黑盒水印嵌入和密码标签生成工具",

    # 你要安装的包，通过 setuptools.find_packages 找到当前目录下有哪些包
    packages=find_packages(),

    # 表明当前模块依赖哪些包，若环境中没有，则会从pypi中下载安装
    install_requires=[
        'numpy>=1.17.0',
        'opencv-python',
        'setuptools',
        'PyWavelets'
    ],

    # install_requires 在安装模块时会自动安装依赖包
    # 而 extras_require 不会，这里仅表示该模块会依赖这些包
    # 但是这些包通常不会使用到，只有当你深度使用模块时，才会用到，这里需要你手动安装
    # extras_require={
    #     'flask': ['flask'],
    #     'flask-marshmallow': ['flask-marshmallow'],
    # }
)