Fix vision.h import not found

tianzhi0549 · Aug 11, 2021 · 9a76a97 · 9a76a97
1 parent 0eb8ee0
commit 9a76a97
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -25,7 +25,8 @@
 
 
 def get_extensions():
-    extensions_dir = os.path.join("fcos_core", "csrc")
+    extensions_dir = os.path.abspath(
+        os.path.join(os.path.dirname(__file__), "fcos_core", "csrc"))
 
     main_file = glob.glob(os.path.join(extensions_dir, "*.cpp"))
     source_cpu = glob.glob(os.path.join(extensions_dir, "cpu", "*.cpp"))