Skip to content

grounding dino1.5 debug #55

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
h1303395658-hub wants to merge 2 commits into
base: master
Choose a base branch
from
Open

grounding dino1.5 debug #55

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
fb006f6
修改上传文件时的日志输出
h1303395658-hub Jan 5, 2026
9b14048
修改上传文件时的日志信息
h1303395658-hub Jan 5, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 13 additions & 3 deletions gdino/model_wrapper.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -120,13 +120,23 @@ def get_image_url(self, image: Union[str, np.ndarray]):
Returns:
str: The url of the image
"""
if isinstance(image, str):
if isinstance(image, str):
# 如果 image 是文件路径
url = self.client.upload_file(image)
else:
# 如果 image 是 numpy 数组，转换为 PNG 格式
with tempfile.NamedTemporaryFile(delete=True, suffix=".png") as tmp_file:
# image is in numpy format, convert to PIL Image
# image 是 numpy 格式，转换为 PIL 图像
image = Image.fromarray(image)
image.save(tmp_file, format="PNG")
tmp_file_path = tmp_file.name
url = self.client.upload_file(tmp_file_path)

# 上传文件并打印响应信息
rsp = self.client.upload_file(tmp_file_path)
print(f"Status code: {rsp.status_code}")
print(f"Response text: {rsp.text}") # 打印返回的响应文本
assert rsp.status_code == 200, f"上传文件失败，状态码: {rsp.status_code}, 响应内容: {rsp.text}"

url = rsp.text # 假设返回的是文本，保存响应内容为 URL

return url