setup.py

import glob
import os

from setuptools import find_packages
from setuptools import setup

setup(
    name="scalelsd",
    version="1.0",
    author="Zeran Ke and Nan Xue",
    description="Scalable Deep Line Segment Detection Streamlined",
    packages=find_packages(),
    install_requires=[
        "torch", 
        "torchvision",
        "accelerate",
        "tensorboard",
        "timm",
        "opencv-python-headless==4.8.1.78",
        "kornia",
        "cython",
        "matplotlib",
        "yacs",
        "scikit-image",
        "tqdm",
        "python-json-logger",
        "h5py",
        "shapely",
        "seaborn",
        "easydict",
    ],
    extras_require={
        "dev": [
            "pycolmap",
        ]
    }
)