在哪里可以找到 YOLOv5 特征提取器？

Question

我必须为学校项目定制 YOLOv5 特征提取器。克隆 YOLOv5 git 存储库后，我看到了一堆文件。但我不知道特征提取器在哪里。有人可以指导我查看一些文件吗？

Answer 1

可以使用

_forward_once

中的

BaseModel

来提取特征。你可以在这里找到它：

https://github.com/ultralytics/yolov5/blob/868c0e9bbb45b031e7bfd73c6d3983bcce07b9c1/models/yolo.py#L114

一些相关讨论：

• https://github.com/ultralytics/yolov5/issues/385
• https://github.com/ultralytics/yolov5/issues/1173

Answer 2

您找到问题的解决方案了吗？