- 博客(2)
- 收藏
- 关注
原创 YOLO V5生成热力图及遇到的问题
把yolov5_object_detector.py文件中的datasets改为dataloaders,即。4. 在models文件夹下添加yolov5_object_detector.py文件,内容如下。先看一下utils文件夹下有没有一个datasets文件,如果没有就是dataloaders文件。3. 在models文件夹下添加gradcam.py文件,内容如下。5. 在根目录下添加main_gradcam.py文件,内容如下。yolov5_object_detector.py文件中。
2023-06-06 15:39:36
1335
6
原创 YOLO V5(6.0)添加CBAM、SENet、CA注意力机制手把手教程
3)在models文件夹下创建yaml文件,命名为yolov5-cbam.yaml。3)在models文件夹下创建yaml文件,命名为yolov5-se.yaml。3)在models文件夹下创建yaml文件,命名为yolov5-ca.yaml。---cfg中添加yaml文件路径(与CBAM添加路径一致)---cfg中添加yaml文件路径(与CBAM添加路径一致)---cfg中添加yaml文件路径。4)在train.py中修改。4)在train.py中修改。4)在train.py中修改。
2023-05-26 10:08:59
3509
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人