- 博客(53)
- 收藏
- 关注
原创 mmdetection 操作记录
或者可以把 mmengine 和 mmdet 的 github 源码下载到 project 文件夹下,这样不需要指令和。个人更建议源码方式,因为方便修改 mmengine 和 mmdet 下的代码。否则是按照 pip 包方式调用,不好修改其中的代码。
2025-12-29 12:05:51
199
原创 Faster RCNN - RPN作用原理
以mmdet中的mmdet/configs/_base_/models/faster_rcnn_r50_fpn.py实现为参考,介绍其中RPN的工作原理。RPN的网络结构:shared conv * 1 + 2 conv heads 相当于对feat map的每个位置进行了cls和bbox预测。
2025-10-05 11:48:50
801
原创 coco&voc数据格式和工具使用
coco是一种常见的目标检测标注格式,按该格式进行标注后可以直接调用pycocotools库进行读取和指标计算,十分方便。
2025-09-02 10:06:37
251
原创 QLoRA操作记录
对于VLM,在vpm(视觉特征提取器)和llm之间会有resampler,一些resampler使用了nn.MultiheadAttention。如果要再加载save_path中的模型参数并进行训练,可以将save_path下的内容复制到base_model文件夹中(该文件夹下包含模型构建代码),再使用上述FM9GV.from_pretrained加载。中加入"resampler",代表不对模型中的self.resampler进行量化。同理,测试发现"lm_head"也不能量化,否则报错。
2025-07-03 21:16:42
246
原创 ubuntu使用Postfix外部SMTP代理发送邮件
ubuntu使用指令发送邮件。普通的mail指令通常被识别为垃圾邮件,且由于ubuntu主机邮箱地址不正规,执行指令几次后常被鉴定为风险地址从而禁止。对此,使用外部SMTP代理,驱动正规地址(例如qq邮箱)向指定邮箱发送。
2025-05-11 21:40:20
676
原创 调试pytorch DDP训练代码
Pytorch提供了Distributed Data Parallel (DDP)工具以便在多机多卡上并行训练,并提供了torchrun指令来启动。然而,torchrun指令启动不便于debug。可以通过修改成等价mp.spawn启动方式先debug,完成后再转回torchrun指令启动正式训练。
2024-12-31 10:33:05
746
原创 【代码模板】Pytorch混合精度训练
当使用AMP混合精度训练时,可以提升训练速度,并降低对显存的占用。下面提供一个使用AMP训练的代码demo。
2024-10-14 21:14:55
649
2
原创 【代码模板】统计数据集的均值和标准差
在数据预处理时,通常会对样本进行标准化操作,使样本的均值为0,标准差为1,从而提高训练的稳定性。进行标准化操作时,需要预先统计数据集的均值和标准差。下面的demo展示了如何实现这个操作。
2024-10-06 10:33:13
394
原创 【代码模板】Python Decorator / 装饰器
在 python 里面,函数是一个对象,但是是 callable 的对象,因此后面可以接 () 传入参数。这也说明函数可以被当作参数传入其他的函数里面,同时函数也可以作为返回值。
2024-09-28 09:20:58
533
原创 【SOP】Windows下安装Neo4j流程
Neo4j 是一个基于图形结构的 NoSQL 数据库,专门用于存储和管理图数据。与传统的关系型数据库不同,Neo4j 使用 图(graph)的形式来表示数据,其中数据点(称为 节点)通过 边(relationships)相互连接。Neo4j使用Java语言开发。
2024-09-22 21:01:46
2967
2
原创 【代码模板】张量操作,文件读写
这里实现了一个功能 demo,输入array 是 (13, 17, 2) float32,代表某一帧的所有人体关键点识别结果,该帧有13个人,每个人识别17个关键点,每个点用 x,y 坐标表示。现在希望根据每个人关键点重心的 x 坐标值,按从小到大的规则,沿第一维对 array 重新排序。
2024-09-12 16:45:39
271
原创 【安装环境】配置MMTracking环境
numpy 的版本不能太高,否则后面安装时会发生冲突。先安装 numpy,因为 pytorch 的安装会自动配置高版本 numpy。mmtracking 支持的 torch 版本有限,需要找到合适的。
2024-09-12 09:11:57
837
原创 C盘太大怎么办
C:\Windows\Installer是一个很大的文件夹,可以移动到其他盘减少C盘负担。第一次寻找 C:\Windows\Installer 可能发现没有该可视化文件夹。如果要恢复该文件夹的位置,删除 symbolic link。
2024-09-11 19:19:49
5720
原创 理解 Progressive GAN
介绍模型 Progressive GAN,PGGAN, 论文《PROGRESSIVE GROWING OF GANS FOR IMPROVEDQUALITY, STABILITY, AND VARIATION》。
2023-01-20 17:28:45
1328
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅