File size: 1,232 Bytes
a9d31d8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
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'
]
}
)
|