update inference

detect_tools/upn/ops/setup.py

@@ -18,7 +18,11 @@ from torch.utils.cpp_extension import CUDAExtension
 from setuptools import find_packages
 from setuptools import setup
 
-requirements = ["torch", "torchvision"]
+setup(
+    # ... 其他配置 ...
+    setup_requires=['torch', 'ninja'] # <-- 添加 setup_requires
+)
+# requirements = ["torch", "torchvision"]
 
 def get_extensions():
     this_dir = os.path.dirname(os.path.abspath(__file__))

requirements.txt

@@ -7,4 +7,5 @@ gradio
 mmengine==0.8.2
 einops
 ninja
-scikit-image
+scikit-image
+-e ./VLM-FO1/detect_tools/upn/ops