From 4080c923558c33907442ec9627ff6ea7120ebff8 Mon Sep 17 00:00:00 2001 From: captainfffsama Date: Wed, 12 Apr 2023 17:30:31 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0dataset=E6=A8=A1=E5=9D=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index a045b76..fc33aeb 100644 --- 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"))