AttributeError: 'Detections' object has no attribute 'imgs'. Did you mean: 'ims'?
时间: 2023-08-22 19:04:00 浏览: 191
AttributeError: module 'tensorflow.compat.v1' has no attribute '
如果你遇到了 `AttributeError: 'Detections' object has no attribute 'imgs'. Did you mean: 'ims'?` 错误,这意味着 `Detections` 对象中没有 `imgs` 属性。该错误信息还建议是否要使用 `ims` 属性。
根据你之前提供的代码,你可能是想使用 `ultralytics/yolov5` 库中的 `Detections` 类的 `ims` 属性。你可以修改代码如下:
```python
import torch
from PIL import Image
from pathlib import Path
# 加载YOLOv5模型
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
# 设置要检测的图片文件夹路径
image_folder = "path/to/your/image/folder" # 替换为你的图片文件夹路径
# 设置保存检测结果的文件夹路径
output_folder = "path/to/your/output/folder" # 替换为你的输出文件夹路径
# 获取文件夹中的所有图片文件
image_files = sorted(list(Path(image_folder).glob("*.jpg")))
# 遍历每张图片进行检测
for image_file in image_files:
# 加载图片
img = Image.open(image_file)
# 进行图像检测
results = model(img)
# 获取图片文件名
file_name = image_file.stem
# 设置保存结果的文件路径
output_file = Path(output_folder) / f"{file_name}_result.jpg"
# 保存检测结果图像
results.ims.save(output_file)
# 输出检测结果的文件路径
print(f"Saved detection results to: {output_file}")
```
在修改后的代码中,我们使用 `results.ims.save(output_file)` 将检测结果图像保存到指定路径 `output_file`。这样,你就可以保存检测结果图像了。请注意,我们将 `results.ims` 替换了 `results.imgs`。
阅读全文