- 博客(27)
- 收藏
- 关注
RSS订阅原创 【目标检测】根据检测模型推理结果构造一个XML文件
目标检测的数据集标注一般通过labelimg插件进行标注，有很多样本类型很容易进行标注对于这些很容易进行标注的数据集我们在进行模型迭代的时候经常也要进行标注这样导致很废时间，我们假设是在你已经有一个训练好基于某个任务的模型，这时你又有了新的训练样本那么如果不想费力再一个一个进行标注，可以通过当前已经训练好的模型线对这些未标记的样本进行推理再将每个样本的结果保存到xml文件中。将保存下来的xml文件使用labelimg插件就可以自动进行匹配。
2023-03-19 16:41:44
244
原创 【目标检测】根据检测模型的推理结果进行预测框的绘制
目前主流的目标检测器以YOLOX为例，后处理后的output基本上output[: , :4]前四个代表的bbox的坐标一般会将中心宽高的形式转换成左上角右下角点坐标的形式，至于output[: , :4] output[:, :5]分别是obj是否包含物体的得分和类别得分，但是目前例如YOLOv8已经没有obj得分了似乎只有分类得分大家根据情况对脚本进行更改。最后一个维度为类别索引.首先这里这里只是给大家提供一个脚本，具体从model出来的output需要经过的一些后处理例如nms自行解决。
2023-03-19 15:41:08
520
原创 【机器学习】利用DBSCAN-基于密度的空间聚类算法计算出一批经纬度中的中心点
【机器学习】利用DBSCAN-基于密度的空间聚类算法计算出一批经纬度中的中心点，并筛选出离群点。
2023-03-07 16:54:13
1387
1
原创 【PIL】OSError: cannot write mode RGBA as JPEG
如果你保存的图片是.png结尾通过重命名的方式更改为jpg会再进去存储的时候会报错，主要因为png图片默认有四个通道。PNG是四通道：RGBA意思是红色，绿色，蓝色，Alpha的色彩空间，Alpha指透明度。JPG是三通道：RGB意思是红色，绿色，蓝色。转换成rgb图片即可。
2023-03-06 16:40:39
568
原创 【pytorch】Pytorch训练好的模型转torchscript
TorchScript，它是PyTorch模型（子类nn.Module）的中间表示，可以在高性能环境（例如C ++）中运行。利用torch.jit.trace()函数。
2023-03-05 15:39:43
1430
4
原创 【PIL】不失帧的resize操作
在cv的项目以目标检测，图像分类为例，都需要把图像resize成固定大小然后传入网络去提取特征，如何对你要训练的数据的一些特性合理的使用resize成为比较关键的前处理因素。
2023-03-05 15:33:10
1176
原创 Repvgg详解及其实现（pytorch）
原论文中的结构图很直观的展示了repvgg到底是什么意思，对比Resnet它仍然有着类似残差的结构，就是在3*3的卷积基础上弄了一个1*1的分支和一个identity分支(在步长不等于2且输入输出通道相等的情况下)，这样除了方便推理过程的融合，似乎还有多分支的好处，因为有丰富的梯度信息(狗头保命)，但是这好像会在训练时增加参数量，但是推理速度快就ok了。这样我们就将三个分支融合成了一个分支，这样的在推理的时候就会非常快，而且很省内存。写这边博客也是为了做一个记录，如果有错误或者不足，请各位大佬提出。
2022-09-30 16:22:40
9000
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人