setup.py

from setuptools import find_packages, setup

import sys


def parse_requirements(filename="requirements.txt"):
    # Read requirements.txt, ignore comments
    try:
        requirement_list = list()
        with open(filename, "rb") as f:
            lines = f.read().decode("utf-8").split("\n")
        for line in lines:
            line = line.strip()
            if "#" in line:
                # remove package starting with '#'
                line = line[: line.find("#")].strip()
            if line:
                if line.startswith("opencv-python"):
                    # in case of conda installed opencv, skip installing with pip
                    try:
                        import cv2

                        print(cv2.__version__)
                        continue
                    except Exception:
                        pass

                if line.startswith("torch"):
                    if sys.platform == "darwin":
                        line = "torch"

                if line.startswith("mmcv"):
                    if sys.platform == "darwin":
                        line = "mmcv"

                requirement_list.append(line)

    except Exception:
        print(f"'{filename}' not found!")
        requirement_list = list()
    return requirement_list


required_packages = parse_requirements()

setup(
    name="med_query",
    version="1.0",
    description="Codebase for Med-Query Framework",
    author="DAMO MIA group",
    packages=find_packages(),
    install_requires=required_packages,
    dependency_links=[],
    include_package_data=True,
    zip_safe=False,
    test_suite="nose.collector",
    tests_require=["nose"],
    entry_points={
        "console_scripts": [
            "med_query_vis=med_query.vis.visual:main",
            "med_query_train_det=med_query.utils.cli_functions:med_query_det_train_dist",
            "med_query_train_seg=med_query.utils.cli_functions:med_query_seg_train_dist",
            "med_query_train_roi=med_query.utils.cli_functions:med_query_seg_train_dist",
            "med_query_test=med_query.det.test:main",
            "med_query_valid=med_query.det.valid:main",
            "med_query_det_metrics=med_query.utils.det_metrics:main",
            "med_query_seg_metrics=med_query.utils.seg_metrics:main",
            "stop_distributed_training=med_query.utils.cli_functions:stop_distributed_training",
        ]
    },
)