- 博客(15)
- 收藏
- 关注
原创 【备份】ncclInvalidUsage: This usually reflects invalid usage of NCCL library
1. CUDA_VISIBLE_DEVICES=0,1,(举个例子)前面的变量名有没有敲错;2. CUDA、nccl、pytorch版本是否兼容,我问了一下chatgpt我的是兼容的;以及其他博主的解决办法>_<
2024-11-05 21:06:19 111
原创 Xshell密钥登录的密码
之前使用ssh创建密钥对的时候不记得有设置什么私钥密码,但Xshell用户身份验证非要输个密码。即可设置一个新密码,其中~\.ssh\id_rsa是存放私钥的路径。然后在Xshell中重新倒入私钥并输入这个密码即可。
2024-07-31 16:52:56 283
原创 【Python】*args,**kwargs 及打包、解包
*把传入的一系列键值变量打包成一个字典,存入变量 kwargs。然后将 "aaa", a, *b 打包作为 args,x=111, y="bbb", **d 打包作为 kwargs,args 和 kwargs 分别为元组和字典;函数调用时,*和**表示解包。*args表示可以传入任意数量的“值变量”,**kwargs表示可以传入任意数量的键值变量。print可以传入*args参数,但无法传入**kwargs,需要使用.format()首先 *b 和 **d 对 b 和 d 解包,打散成元素和键值;
2024-01-10 10:24:15 555
原创 【笔记】YOLOv7辅助检测Aux+损失函数
辅助检测头主要用于训练时,可以提升召回率,训练更好,但同时训练成本也会增加,预测时使用没有辅助头的普通检测头。总结一下目前YOLOv7代码,加了辅助检测分支的脚本和没加的区别。使用Aux的模型文件位于,同时其对应的deploy时的模型位于。二者只有在head部分不一样,训练时最后一层是IAuxDetect,预测时是普通的Detect。加不加辅助头的训练脚本也分开了,常规的是train.py,带辅助头的是train_aux.py,两个脚本。
2023-11-22 20:45:37 3500
原创 【笔记】YOLOv7重参数化(RepConV)原理+代码
在detect时,将部分结构的参数进行融合,来达到提高推理速度而不损失精度的目的,前提是已经训练好了模型参数。该部分内容可以通过 detect.py/detect()函数中进入attempt_load(),中的fuse()函数。该函数位于主要看。(m是IDetect时的融合是隐式implicit部分的融合,非本文重点)
2023-11-20 20:33:23 6624 1
原创 【笔记】YOLOv7正样本匹配原理+代码(SimOTA)
前两步筛选出的正样本只是anchor,没有预测值也可以获得,目的是为了增加正样本数量;第三步SimOTA精细化筛选,涉及预测值,需要经过前向传播得到pred后才能计算,第1,2步的正样本anchor,给第3步提供pred索引进行初筛,再经过cost筛选,此时的“正样本”可以直接认为是接下来用来计算loss的预测框了。
2023-11-18 21:32:24 1078 2
原创 【Pytorch】None用于扩展维度
使用冒号时,None在第几位,就在那一维上扩展一个维度;如果要扩展多个维度,把冒号视作原始维度,None插入所需的位置即可,最后一个None后面的冒号可以省略不写。使用省略号,在最后一维扩展维度。
2023-10-12 19:23:29 240
原创 【备份】Win11安装WSL并打开VSCode
安装完后,会发现开始界面多了一个ubuntu。Error:无法解析服务器的名称或地址尝试更改DNS再重试:打开控制面板--网络和Internet--网络和共享中心:勾选上Internet协议版本4,并双击打开,更改如下:1、百度:180.76.76.762、阿里云:223.6.6.6 或 223.5.5.53、腾讯:119.29.29.294、 电信:101.226.4.65、联通:123.125.81.66、移动:101.226.4.67、铁通:101.226.4.6。
2023-09-21 16:43:41 672
原创 【自用】YOLO v2原理总结
个人认为,YOLO v2相比v1最大的改进在于引入了anchor,然后改用了DarkNet-19网络结构,并且在卷积层后都加入了BN层。速度较之更快,mAP也更高。参考博客:YOLOV2网络模型_yolov2网络结构_Pywin的博客-CSDN博客目标检测|YOLOv2原理与实现(附YOLOv3) - 知乎 (zhihu.com)YOLO v2 损失函数源码分析 - 一只有恒心的小菜鸟 - 博客园 (cnblogs.com)【机器学习】一文带你深入全局平均池化 - 知乎 (zhihu.com)
2023-09-09 16:45:34 259
原创 【自用】YOLO v1总结
对于负责预测物体的grid cell(obj真值框中心落在该grid cell内),2个bbx的x,y都为obj中心点相对于该grid cell左上角的偏移量,2个bbx的w,h都为真值框宽高相对于图像宽高的比值,2个bbx的conf都设为1,有了预测框后改为真值框分别与2个预测框的IOU;③ 从第一个框开始设为bbx_max,计算它与其后面的所有非0框的IOU,IOU大于阈值的设score=0(即重合度很高的两个bbx,取score大的那个)筛去与其重合度高的框。没考虑的部分损失忽略了。
2023-09-08 11:07:34 114
原创 ValueError: setting an array element with a sequence. The requested array has an inhomogeneous shape
在使用np.asarray()或np.asanyarray()时报这个错,是因为要array化的对象“在第一维后具有不均匀的形状”
2023-08-25 11:52:06 36466 4
原创 【备份自用】Eigen库中欧拉角、方向余弦矩阵、四元数、旋转矢量之间的转换
简单整理一下c++ Eigen库中欧拉角、方向余弦矩阵、四元数、旋转矢量之间的转换。
2023-07-14 11:11:41 1048 2
原创 【备份自用】win11双系统删Ubuntu
很久之前电脑被装了Ubuntu16.04,每次开机都要选系统,磁盘空间也越来越小,鉴于目前虚拟机能满足小需求,而且也基本用不到16.04版本,就想自己动手删掉,以清一下内存。不一定是最简单有效的方法,但达成了我的目的。
2023-07-14 10:59:43 1225
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人