yolov5出现报错,原因是网络层
解决的办法:从github上面下载yolov5的源码,ultralytics发布的ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)
打开v5/model/yolo.py文件
复制三个类:Segment(Detect)、BaseModel(nn.Module)、DetectionModel(BaseModel)
yolov5出现报错,原因是网络层
解决的办法:从github上面下载yolov5的源码,ultralytics发布的ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite (github.com)
打开v5/model/yolo.py文件
复制三个类:Segment(Detect)、BaseModel(nn.Module)、DetectionModel(BaseModel)