build: fix install of detectron2 b constraining setuptools

Author: fracape

pyproject.toml

@@ -1,5 +1,5 @@
 [build-system]
-requires = ["setuptools>=68", "wheel", "pybind11>=2.6.0"]
+requires = ["setuptools>=68,<81", "wheel", "pybind11>=2.6.0"]
 build-backend = "setuptools.build_meta"
 
 [project]
@@ -36,7 +36,7 @@ dependencies = [
     "ptflops>=0.7.3",
     "Cython==3.0.8",
     "wheel>=0.45.1",
-    "setuptools>=68",
+    "setuptools>=68,<81",
     "yuvio",
 ]
 

scripts/install.sh

@@ -189,7 +189,7 @@ run_prepare() {
 
 run_install () {
     detect_env
-    "${PIP[@]}" install -U pip wheel setuptools
+    "${PIP[@]}" install -U pip wheel "setuptools>=68,<81"
 
     echo "Selected Models to be installed: ${MODEL}"
 
@@ -212,7 +212,7 @@ run_install () {
         "${PIP[@]}" install -e "${COMPRESSAI_VISION_ROOT_DIR}/compressai"
     elif [[ "${PACKAGE_MANAGER}" == "uv" ]]; then
         echo "Building compressai C++ extensions from source..."
-        "${PIP[@]}" install "pybind11>=2.6.0" "setuptools>=68" wheel setuptools
+        "${PIP[@]}" install "pybind11>=2.6.0" "setuptools>=68,<81" wheel
         cd "${COMPRESSAI_VISION_ROOT_DIR}/compressai"
         rm -rf build/ **/*.so
         "${PIP[@]}" install -e . --no-build-isolation
@@ -291,6 +291,7 @@ install_detectron2 () {
 
     cd "${MODELS_SOURCE_DIR}/detectron2"
     cp ${SCRIPT_DIR}/install_utils/detectron2_pyproject.toml ./pyproject.toml
+    git apply "${SCRIPT_DIR}/install_utils/patches/0002-detectron2-lazy-torch-import.patch" || echo "Patch could not be applied. Possibly already applied."
 
     if [[ "${PACKAGE_MANAGER}" == "pip3" ]]; then
         "${PIP[@]}" install --no-build-isolation -e .

scripts/install_utils/detectron2_pyproject.toml

@@ -37,9 +37,10 @@ requires = [
     "grpcio>=1.48.2",
     "markdown>=2.6.8",
     "protobuf!=4.24.0,>=3.19.6",
-    "setuptools>=41.0.0",
+    "setuptools>=68,<81",
+    "wheel",
     "tensorboard-data-server<0.8.0,>=0.7.0",
     "werkzeug>=1.0.1",
     "MarkupSafe>=2.1.1",
 ]
-build-backend = "setuptools.build_meta"
+build-backend = "setuptools.build_meta"

scripts/install_utils/patches/0002-detectron2-lazy-torch-import.patch

@@ -0,0 +1,67 @@
+--- a/setup.py
++++ b/setup.py
+@@ -5,13 +5,17 @@
+ import os
+ import shutil
+ from os import path
++import sys
+ from setuptools import find_packages, setup
+ from typing import List
+-import torch
+-from torch.utils.cpp_extension import CUDA_HOME, CppExtension, CUDAExtension
+ 
+-torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]
+-assert torch_ver >= [1, 8], "Requires PyTorch >= 1.8"
++
++def get_torch():
++    import torch
++
++    torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]
++    assert torch_ver >= [1, 8], "Requires PyTorch >= 1.8"
++    return torch, torch_ver
+ 
+ 
+ def get_version():
+@@ -44,6 +48,8 @@
+     main_source = path.join(extensions_dir, "vision.cpp")
+     sources = glob.glob(path.join(extensions_dir, "**", "*.cpp"))
+ 
++    torch, torch_ver = get_torch()
++    from torch.utils.cpp_extension import CUDA_HOME, CppExtension, CUDAExtension
+     from torch.utils.cpp_extension import ROCM_HOME
+ 
+     is_rocm_pytorch = (
+@@ -144,6 +150,24 @@
+     "detectron2.projects.panoptic_deeplab": "projects/Panoptic-DeepLab/panoptic_deeplab",
+ }
+ 
++BUILD_COMMANDS = {
++    "build",
++    "build_ext",
++    "bdist_wheel",
++    "editable_wheel",
++    "develop",
++    "install",
++}
++SHOULD_BUILD_EXTENSIONS = any(cmd in BUILD_COMMANDS for cmd in sys.argv[1:])
++
++if SHOULD_BUILD_EXTENSIONS:
++    torch, _ = get_torch()
++    ext_modules = get_extensions()
++    cmdclass = {"build_ext": torch.utils.cpp_extension.BuildExtension}
++else:
++    ext_modules = []
++    cmdclass = {}
++
+ setup(
+     name="detectron2",
+     version=get_version(),
+@@ -203,6 +227,6 @@
+             "flake8-comprehensions",
+         ],
+     },
+-    ext_modules=get_extensions(),
+-    cmdclass={"build_ext": torch.utils.cpp_extension.BuildExtension},
++    ext_modules=ext_modules,
++    cmdclass=cmdclass,
+ )