半夜打老虎

摘要： 参考链接 如何计算模型以及中间变量的显存占用大小： https://oldpan.me/archives/how-to-calculate-gpu-memory 如何在Pytorch中精细化利用显存： https://oldpan.me/archives/how-to-use-memory-pyto 阅读全文

摘要： python脚本运行时经常出现一些警告信息，大量的警告信息会对输出信息造成干扰，可以通过以下方式忽略。 例子 忽略警告 在文件开头引入warnings库 import warnings warnings.filterwarnings("ignore", category=Warning) 阅读全文

摘要： 打开python终端，输入： import site site.getsitepackages() 输出如下： 阅读全文

摘要： def drop_cycle(edge, max_length=20): """ 删除列表中形成的小闭环 @edge: 原始顶点id @max_length: 容许闭环的最小长度 return: 输出删除小闭环后的列表 """ drop_list = [] drop_count = 0 for i, 阅读全文

摘要： 最近需要安装一些库，但安装速度很慢且很容易断开，这里记录一下pip安装库的时候使用清华镜像。 清华镜像地址 https://pypi.tuna.tsinghua.edu.cn/simple 使用 假设要安装vedo库，则在原来安装语句上添加-i和镜像地址即可。 pip install vedo -i 阅读全文

摘要： 最近用到了矩方法对目标检测的结果进行后处理，包括求中心点，偏角和长短轴。基本步骤如下： 第一步：将检测到物体的平行框位置作为ROI区域； 第二步：对ROI区域进行预处理（灰度化、二值化）； 第三步：利用矩方法求出物体的矩（OPENCV::moments）； 第四步：利用矩分别计算中心点，偏角和长短轴 阅读全文

摘要： 问题：最近遇到一个小问题，需要对列表中的元素分组，保证每组元素的和尽可能平衡，最后返回每组的值和所对应的下标，这里对处理办法进行记录，方便以后查看。 解决思路： step1: 对列表进行排序，新建输出的空列表out_list[N]; step2: 根据需要分组的个数$N$，每次遍历$N$个元素； s 阅读全文

摘要： 一、C++结构体排序 我将结构体 $result$ 存入在一个vector ${vResult}$ 中，最近需要在C++中按照结构体中的 $score$ 进行排序，在网上查找了一些资料，这里对采用的方法记录一下，方便以后使用。 1、引入头文件 #include <algorithm> 2、定义排序方 阅读全文

摘要： 在Mask Rcnn中传给模型训练的是物体的mask图，但训练中会利用到包含物体的平行框，查看代码后发现gt_box是由mask生成的，其主要思路是找出平行框左上角和右下角的坐标。mask图是一个二值图，利用numpy库的where()和any()函数就可以找出$x_1$和$x_2$以及$y_1$和 阅读全文

摘要： 经常遇到判断平面向量和点的位置关系，做一个简单的记录，方便后面使用。原理是向量的叉乘。 两个同起点的向量A(xA ,yA)和B(xB ,yB)的叉乘公式为： crossV = xA *yB - yA*xB。 展开后即可得到代码中公式，进而可以判断点和向量的位置关系。 一、定义结构体 二、定义判断函数 阅读全文