- 博客(1)
- 资源 (92)
- 问答 (1)
- 收藏
- 关注
RSS订阅
3601
1rknn yolov5 rtsp
rknn yolov5的官方网站
https://github.com/airockchip/rknn-toolkit2
里面有rk3588 yolov5的推理示例程序，也包括Android推理实例程序，但是Android平台不支持rstp实时推理
本项目用ffmpeg拉流，mpp硬解码，然后rknn yolov5推理跑通，
调用代码在MainActivity，识别结果mpp硬编码保存为264视频文件。
本项目也支持264裸流视频文件推理。
2024-04-21
问财选股导出excel pythony源代码
python问财选股导出excel功能开发，导出csv格式，带界面，也可以基于这个做二次开发或者改进
依赖项pandas，pyqt5
2024-04-21
自研lstm多目标跟踪，DanceTrack验证集精度98%
多目标跟踪这几年一直是算法难题，用过传统跟踪算法，fairmot、bytetrack，效果不是特别理想，自己思考研究了一种基于lstm和多头注意力机制的多目标跟踪算法，支持gru，tcn，TimeSeriesTransformer，lstm和多头注意力机制效果最好，在DanceTrack数据集验证集精度99%，其他算法精度也在98以上， 1060显卡每帧平均推理速度2ms。
本算法是多目标跟踪算法，算法不含检测，检测算法可以用任意检测算法对接。
生成训练数据：
get_track_data_new.py
可视化数据标签：
data/view_track.py
训练入口：
train_cos_lstm.py
可视化测试入口：
demo_track.py
测试准确率：
demo_track_cls.py
本算法算是在多目标跟踪算法的一种思考，抛转引玉，欢迎有志之士一起交流分享。
2024-04-18
强大好用的人体关键点标注工具
之前想研发一款好的人体关键点识别算法，发现网上竟然没有一款好用的人体关键点标注工具，labelme标注时各个关节点是断开的，不能联动，然后自己就开发了一款人体关键点标注工具，关节点直接可以联动，还有关节点放大特效，方便标注。
实际效果演示网址：
https://www.bilibili.com/video/BV18A4m1G73y/?vd_source=2cd2b3646627adc5dc5f2bf8b7fac8fd
https://blog.csdn.net/jacke121/article/details/135849963
2024-04-18
pyopengl windows x64 whl安装包
pyopengl windows x64 whl安装包
本资源提供了python3.9和3.10版本，x64位安装包。
官方的下载地址已经失效
http://www.lfd.uci.edu/~gohlke/pythonlibs/#pyopengl
2024-02-17
mmpose 预测转json标注
mmpose 预测的2d点转成json文件，方便labelme标注
背景：mmpose或者其他2d关键点，bad case标注是个难题，这个脚本可以把mmpose预测的关键点，转为json文件，方便进行二次标注或修改。
2024-01-25
3d-nms cuda代码
3d_nms c++代码和 cuda cu代码，win11编译成功,linux也可以用
python setup.py build
python setup.py install
2024-01-07
MobileViT v3 pytorch代码
MobileViT v3 pytorch代码，几乎都不能加载官方的预训练模型，本人经过调整修改，
可以加载官方预训练模型，原创不易，希望多多支持
2023-12-31
OpenPCDet ingroup-inds库 windows编译
OpenPCDet ingroup_inds库 windows编译
原版是linux代码，在Windows上编译失败，本人经过修改，可以编译成功。
2023-12-27
Ultra-Fast-Lane-Detection-v2 pytorch版
Ultra-Fast-Lane-Detection-v2 pytorch版
Ultra-Fast-Lane-Detection-v2官方原版数据加载用的nvidia.dali.pipeline
Windows没有nvidia库，所以无法训练，数据增强也不方便，特地实现了pytorch的dataset版，可以训练，推理预测
训练收敛更快
可以在这个基础上改进，很方便。
2023-10-20
insightface/alignment 人脸关键点的pytorch版本
landmark 106个点，很强大的算法，算法介绍；
https://blog.csdn.net/jacke121/article/details/116605765
模型4.8m，
gpu测试13ms左右，分辨率192*192
入口代码：
demo_v5_face.py
学习研究比较合适
2023-04-29
yolov5 face onnx推理代码
yolov5 face onnx推理，支持原版yolov5 face 导出的onnx，亲测成功
支持yolov5 face m模型，也支持yolov5 blazeface 。
2023-03-26
yolov7 分割 tensorrt 推理源代码
yolov7 mask分割 tensorrt 推理源代码，网上没有人实现，和大佬一起熬了好几个通宵研发出来，支持tensorrt8，win10平台，亲测成功，可运行
linux也支持，可自行跨平台适配。
2023-02-23
labelme-5.1.1
labelme-5.1.1，设置了自动保存，标签不保存图片，自定义标签颜色，亲测成功，
博客地址：
https://blog.csdn.net/jacke121/article/details/112288397
2023-01-27
3DDFA-V2.5-opencv-yolov5face
3DDFA_V2.5原本人脸检测器是facebox，
facebox在脸部超过一半被遮挡时效果较差，
人脸检测器换成yolov5-face，yolov5-face在脸遮挡超过一半时关键点不太准确。
图片处理是imageio，换成opencv。
demo_cam_v5face.py
yolov5-face 依赖项：
models，加载模型用
yolov5_face，人脸检测器。
2022-12-25
tensorrt c++ 推理项目
tensorrt c++ 推理示例项目，支持分类网络，比如alexnet，mobileone,skipnet等轻量级网络。
环境：win10 vs2017环境，cuda11.0，亲测成功。
tensorrt库版本：
TensorRT-8.2.1.8.Windows10.x86_64.cuda-11.4.cudnn8.2
转换流程：pytorch转onnx，onnx转tensorrt引擎
python生成tensorrt引擎步骤，可以查看我的博客：
https://blog.csdn.net/jacke121/article/details/125382721
2022-07-03
yolov5各个版本 tensorrt cfg文件
yolov5各个版本 tensorrt cfg文件，下载地址：
https://github.com/enazoe/yolo-tensorrt/blob/master/yolov5_tutorial.md
担心作者哪天不再共享，特地下载下来，分享给大家。
2022-02-16
coco2017转xml格式，精选人体数据集
coco2017转xml格式代码，并且手工挑选，去掉超级难度样本，留下精选样本列表
比如只有一只手，一只脚的样本；
剔除远处的超级样本，人体像素小于20个像素的超级小人体。
本人手工标注，花了3个周的时间，原创不易，请珍惜。
2022-01-13
Yolov5_DeepSort_bytetrack
Yolov5 DeepSort跟踪 bytetrack跟踪实例代码，
deepsort 跟踪示例代码：track.py
ByteTrack超越了此前所有的跟踪方法。
参考资料：https://blog.csdn.net/zhouchen1998/article/details/120932206
bytetrack示例代码：track_byte_demo.py
2021-11-20
yolov5 代码版 芒果检测 带训练权重，可以直接用
yolov5 代码版 芒果检测
关于代码版，不是配置文件版，修改网络非常方便，参见我的博客：
https://blog.csdn.net/jacke121/article/details/118392154
芒果检测 带训练权重，可以直接用
2021-11-16
旋转目标检测标签转换为yolov5格式
旋转目标检测标签转换，支持以下数据格式转换为yolov5 txt 模式：
dota 数据格式转换，
itc_vd数据格式转换
uavrod_xml 数据格式转换
ucas数据格式转换
训练代码：
https://github.com/BossZard/rotation-yolov5
2021-11-13
yolov5-face_align.rar
yolov5-face 对齐人脸，yolov5-face 人脸检测，检测5个关键点，并对齐转正。
入库：detect_align.py
2021-09-01
256_256_resfcn256_weight.index
PRNet_Mask 的预训练模型只开源了256_256_resfcn256_weight.data-00000-of-00001，但是pytorch加载预训练时，缺256_256_resfcn256_weight.index，自己测试成功，大家可以方向下载。
网络代码开源地址：
https://github.com/YadiraF/PRNet
https://github.com/bleakie/MaskInsightface/tree/master/PRNet_Mask
2021-08-21
yolov5代码版，带预训练，测试脚本，可训练
yolov5配置版，阅读修改起来不太方便，特地整理出来代码封装版，阅读，修改网络比较方便，带预训练权重，检测代码：detect_class_s.py，检测芒果的，可以训练别的。
2021-08-21
神经网络改进的几个思考
2022-04-09
TA创建的收藏夹 TA关注的收藏夹
TA关注的人