首页 > 解决方案 > 我在使用 jetson nano 上的传感器进行面罩检测程序中使用 yolov5 的自定义权重,运行后出现错误

问题描述

正在下载https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt到 /root/.cache/torch/hub/ultralytics_yolov5_master/yolov5s.pt... 100%|#### ##################################| 14.1M/14.1M [00:05<00:00, 2.72MB/s]

融合层...模型摘要:224 层,7266973 个参数,0 个渐变添加自选图形...地址处没有 I2C 设备:0x69 回溯(最后一次调用):文件“maskcamera.py”,第 224 行,在 identify_maskman(0 ) 文件“maskcamera.py”,第 185 行,在 identify_maskman 预测 = model.predict(data) 文件“/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”,行948,在getattr 类型(自我)中。name , name)) AttributeError: 'AutoShape' object has no attribute 'predict' Exiting... 清理引脚

标签: pythonnvidia-jetson-nanoyolov5

解决方案


推荐阅读