树莓派4B基于PyTorch的目标检测遇ModuleNotFoundError求助

1. 修复torchvision依赖问题 #

错误提示'torchvision.transforms' is not a package是因为torchvision未正确安装或版本不兼容，树莓派ARM架构需要安装适配的torchvision版本：

• 先卸载现有torchvision：

  pip uninstall torchvision -y
  

• 安装与当前torch版本匹配的torchvision（以torch 2.0.0为例，对应torchvision 0.15.1）：

  pip install torchvision==0.15.1 --index-url https://download.pytorch.org/whl/cpu
  

2. 修正代码中的错误 #

你的代码存在重复导入、缩进错误以及多余导入的问题，修正后的完整代码如下：

import torch
import matplotlib.pyplot as plt
import numpy as np
import cv2
import uuid
import os
import time

# 加载YOLOv5自定义模型
model = torch.hub.load('ultralytics/yolov5', 'custom', path='last1.pt', force_reload=True)
# 强制使用CPU（树莓派默认无GPU）
model.to('cpu')

# 初始化摄像头
cap = cv2.VideoCapture(0)

while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break  # 摄像头读取失败时退出循环

    # 执行目标检测
    results = model(frame)

    # 渲染检测结果并转换格式
    rendered_results = np.squeeze(results.render())
    rendered_results = rendered_results.astype(np.uint8)

    # 显示检测画面
    cv2.imshow('YOLO', rendered_results)

    # 按下q键退出
    if cv2.waitKey(10) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()

3. 额外注意事项 #

• 树莓派CPU性能有限，建议使用YOLOv5s或更小的模型，避免检测速度过慢
• 如果摄像头无法读取，检查摄像头是否正确连接，或者尝试将cv2.VideoCapture(0)改为cv2.VideoCapture(1)
• 确保last1.pt模型文件路径正确，放在代码同目录下或使用完整路径

内容的提问来源于stack exchange，提问作者youngharshil