YOLO
yolo相关算法
粼粼淇
淇水汤汤,波光粼粼。岸芷汀兰,郁郁青青。
从事计算机视觉,机器学习等方面算法的工作,爱好数学、算法、编程、中国古典文学、哲学,喜欢旅游,爱思考、爱推理;追求卓越、追求极致。
展开
-
加载yolov5 模型到cuda0和加载resnet50模型到cuda1出错处理
先说下我工程:我需要使用yolov5的模型给出一张图中的检测结果,然后把这些检测结果再次送到resnet50的分类器中进行分类,刚开始我把两个模型都加载到了cuda0,一切完美运行.为了加速,我把yolov5的模型加载到了cuda0上(说明下我有4块gpu),resnet50的分类器模型加载到了cuda1上.想法是美好的,但执行上述流程时出现了意想不到的错误:RuntimeError: CUDA error: invalid device ordinalCUDA kernel errors mig原创 2022-01-06 17:50:40 · 1595 阅读 · 0 评论 -
IoU、GIoU、DIoU、CIoU
1 IoU、GIoU、DIoU、CIoU损失函数的那点事儿https://zhuanlan.zhihu.com/p/947992952 目标检测算法之AAAI 2020 DIoU Loss 已开源(YOLOV3涨近3个点)https://cloud.tencent.com/developer/article/1558533转载 2021-03-08 16:47:03 · 111 阅读 · 0 评论 -
darknet-yolov4 python接口封装
class DarknetYolov4: def __init__(self,configPath,weightPath,metaPath,thresh): # Import the global variables. This lets us instance Darknet once, then just call performDetect() again without instancing again global metaMain, netMain, a.原创 2020-10-13 13:41:48 · 680 阅读 · 0 评论 -
冻结网络层训练的解释
博客链接:https://zhuanlan.zhihu.com/p/41313280在微调(fine-tuning)中,需要确定冻结的层数和可训练的层数,主要取决于,数据集相似度和新数据集的大小。原则上,相似度越高,则固定(fix)的层数越多;新数据集越大,不考虑训练时间的成本,则可训练更多的层数。然后可能也要考虑数据集本身的类别间差异度,但上面说的规则基本上还是成立的。例如,在图片分类的网络中,底层一般是颜色、轮廓、纹理等基础结构,显然大部分问题都由这些相同的基础结构组成,所以可以冻结这些层。层数转载 2020-05-17 13:58:43 · 4364 阅读 · 2 评论 -
darknet训练yolov3时文件名注意点
1、文件名含有image,raw这样的单词我有一个这样的图片名:imagestore20161022f8b61d26-c58a-41e6-85fa-c4c78d66.jpg当转化为darknet训练yolov3的数据格式时出现了如下问题:labelstore20161022f8b61d26-c58a-41e6-85fa-c4c78d66.txt找不到我有一个这样的图片名:15278...原创 2019-12-25 10:45:32 · 392 阅读 · 0 评论 -
自定义darknet YOLO-V3训练过程输出信息到log文件和模型保存间隔
YOLO-V3可视化训练过程中的参数,绘制loss、IOU、avg Recall等的曲线图参考这篇文章:https://blog.csdn.net/qq_34806812/article/details/81459982?utm_source=blogxgwz3文章很详细,但说实话这个过程感觉有些麻烦,打印信息一大堆,比如我只想看看loss或者agv_loss,这很不简洁,还好dar...原创 2019-10-12 14:15:54 · 9342 阅读 · 13 评论