yolo系列
改进yolo
给我来一杯冰可乐
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
YOLOv5用TensorBoard可视化结果解读
TP为正确识别的判例个数,FN为有目标但识别为没有目标的判例个数。FP为没有目标但识别为有目标的判例个数。原创 2022-10-18 15:19:25 · 6119 阅读 · 1 评论 -
关于yolov5训练时报错cuda内存不足问题
当时报这种错误,当然调小batch-size可以解决,但是当不想调小的时候可以把工作线程改小,就能正常训练了。在train中有如上图所示的超参数,之前我的工作线程是8,在同等条件下改为4就正常能训练了。原创 2022-10-14 20:58:32 · 3529 阅读 · 2 评论 -
划分数据集及格式转换
【代码】划分数据集及格式转换。原创 2022-10-06 15:41:51 · 507 阅读 · 0 评论 -
解决yolov5不能使用 tensorboard --logdir=runs/train的问题
ValueError: Duplicate plugins for name projector这是 报错类型解决这个的方法就是删除一个文件找到你自己的环境,然后找lib文件夹,然后找文件夹site-packages,然后再此文件夹下搜索tensorboard,会看到我类似下边这个图删除这个文件夹tensorboard-2.8.0.dist-info(版本可能不一样,没关系,删除你自己那个就行)就可以了...原创 2022-05-10 21:33:49 · 2868 阅读 · 2 评论 -
Yolov5查看各层特征图方法
使用方法很简单,只需要在utils中的general.py或者plots.py添加如下函数:import matplotlib.pyplot as pltfrom torchvision import transforms def feature_visualization(features, model_type, model_id, feature_num=64): """ features: The feature map which you need to visua原创 2022-04-22 19:13:20 · 8515 阅读 · 21 评论 -
yolov5-5.0加入CBAM,SE,CA,ECA注意力机制
CBAM注意力yolo.py和yaml文件中相应的CBAMC3也要换成CBAM,下面的SE同理class ChannelAttention(nn.Module): def __init__(self, in_planes, ratio=16): super(ChannelAttention, self).__init__() self.avg_pool = nn.AdaptiveAvgPool2d(1) self.max_pool = nn.原创 2022-04-21 13:53:41 · 9523 阅读 · 16 评论
分享