Skip to content

Commit 9a76a97

Browse files
committed
Fix vision.h import not found
1 parent 0eb8ee0 commit 9a76a97

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

setup.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,8 @@
2525

2626

2727
def get_extensions():
28-
extensions_dir = os.path.join("fcos_core", "csrc")
28+
extensions_dir = os.path.abspath(
29+
os.path.join(os.path.dirname(__file__), "fcos_core", "csrc"))
2930

3031
main_file = glob.glob(os.path.join(extensions_dir, "*.cpp"))
3132
source_cpu = glob.glob(os.path.join(extensions_dir, "cpu", "*.cpp"))

0 commit comments

Comments
 (0)