- 再运行YOLOV5界面的文件时,出现
- 解决的办法:从github上面下载yolov5的源码,ultralytics发布的ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)
打开models/yolo.py
复制三个类:Segment(Detect)、BaseModel(nn.Module)、DetectionModel(BaseModel)
复制的内容要把三个类的方法和属性全部复制到自己的yolo.py中,为了方便记住,最好布置注释
此时还有一个错误
没有这个类,跳转到yolov5源码的common.py文件中,找到Proto,复制到自己的common.py中
再运行yolo文件
网络层问题解决