| from setuptools import setup, find_packages | |
| setup( | |
| name='debug', | |
| version='0.1.0', | |
| packages=find_packages(), | |
| install_requires=[ | |
| 'torch==1.11.0+cu113', | |
| 'torchvision==0.12.0+cu113', | |
| 'mmcv==2.0.0rc4', | |
| 'albumentations==1.3.0', | |
| 'fvcore==0.1.5.post20221221', | |
| 'gradio==3.29.0', | |
| 'h5py==3.8.0', | |
| 'imageio==2.27.0', | |
| 'matplotlib==3.7.1', | |
| 'numpy==1.24.2', | |
| 'opencv_contrib_python==4.7.0.72', | |
| 'opencv_python==4.7.0.72', | |
| 'opencv_python_headless==4.7.0.72', | |
| 'Pillow==9.5.0', | |
| 'pycocotools==2.0.6', | |
| 'pyequilib==0.3.0', | |
| 'scikit-learn==1.2.2', | |
| 'scipy==1.9.1', | |
| 'setuptools==59.5.0', | |
| 'timm==0.6.13', | |
| 'tqdm==4.65.0', | |
| ], | |
| dependency_links=[ | |
| 'git+https://github.com/facebookresearch/detectron2.git@4e447553eb32b6e3784df0b8fca286935107b2fd', | |
| 'git+https://github.com/jinlinyi/PerspectiveFields.git@dev#egg=perspective2d' | |
| ], | |
| extras_require={ | |
| 'cu113': [ | |
| '--extra-index-url https://download.pytorch.org/whl/cu113', | |
| '--find-links https://download.openmmlab.com/mmcv/dist/cu113/torch1.11/index.html' | |
| ] | |
| } | |
| ) | |