添加dataset模块

2 years ago · 4080c92355
parent 4f04acf4c8
commit 4080c92355
1 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@ -55,7 +55,7 @@ torch_ver = [int(x) for x in torch.__version__.split(".")[:2]]

 def get_extensions():
    this_dir = os.path.dirname(os.path.abspath(__file__))
-    extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc")
+    extensions_dir = os.path.join(this_dir, "groundingdino", "models", "GroundingDINO", "csrc","datasets")

    main_source = os.path.join(extensions_dir, "vision.cpp")
    sources = glob.glob(os.path.join(extensions_dir, "**", "*.cpp"))