BIWI vocaset模板
BIWI vocaset模板
包括FLAME_sample.ply
FLAME_masks.pkl
BIWI/templates.pkl
BIWI/F1.obj
BIWI.ply
rknn yolov5 rtsp
rknn yolov5的官方网站
https://github.com/airockchip/rknn-toolkit2
里面有rk3588 yolov5的推理示例程序,也包括Android推理实例程序,但是Android平台不支持rstp实时推理
本项目用ffmpeg拉流,mpp硬解码,然后rknn yolov5推理跑通,
调用代码在MainActivity,识别结果mpp硬编码保存为264视频文件。
本项目也支持264裸流视频文件推理。
问财选股导出excel pythony源代码
python问财选股导出excel功能开发,导出csv格式,带界面,也可以基于这个做二次开发或者改进
依赖项pandas,pyqt5
自研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
本算法算是在多目标跟踪算法的一种思考,抛转引玉,欢迎有志之士一起交流分享。
强大好用的人体关键点标注工具
之前想研发一款好的人体关键点识别算法,发现网上竟然没有一款好用的人体关键点标注工具,labelme标注时各个关节点是断开的,不能联动,然后自己就开发了一款人体关键点标注工具,关节点直接可以联动,还有关节点放大特效,方便标注。
实际效果演示网址:
https://www.bilibili.com/video/BV18A4m1G73y/?vd_source=2cd2b3646627adc5dc5f2bf8b7fac8fd
https://blog.csdn.net/jacke121/article/details/135849963
pyopengl windows x64 whl安装包
pyopengl windows x64 whl安装包
本资源提供了python3.9和3.10版本,x64位安装包。
官方的下载地址已经失效
http://www.lfd.uci.edu/~gohlke/pythonlibs/#pyopengl
mmpose 预测转json标注
mmpose 预测的2d点转成json文件,方便labelme标注
背景:mmpose或者其他2d关键点,bad case标注是个难题,这个脚本可以把mmpose预测的关键点,转为json文件,方便进行二次标注或修改。
3d-nms cuda代码
3d_nms c++代码和 cuda cu代码,win11编译成功,linux也可以用
python setup.py build
python setup.py install
MobileViT v3 pytorch代码
MobileViT v3 pytorch代码,几乎都不能加载官方的预训练模型,本人经过调整修改,
可以加载官方预训练模型,原创不易,希望多多支持
OpenPCDet ingroup-inds库 windows编译
OpenPCDet ingroup_inds库 windows编译
原版是linux代码,在Windows上编译失败,本人经过修改,可以编译成功。
Ultra-Fast-Lane-Detection-v2 pytorch版
Ultra-Fast-Lane-Detection-v2 pytorch版
Ultra-Fast-Lane-Detection-v2官方原版数据加载用的nvidia.dali.pipeline
Windows没有nvidia库,所以无法训练,数据增强也不方便,特地实现了pytorch的dataset版,可以训练,推理预测
训练收敛更快
可以在这个基础上改进,很方便。
insightface/alignment 人脸关键点的pytorch版本
landmark 106个点,很强大的算法,算法介绍;
https://blog.csdn.net/jacke121/article/details/116605765
模型4.8m,
gpu测试13ms左右,分辨率192*192
入口代码:
demo_v5_face.py
学习研究比较合适
yolov5 face onnx推理代码
yolov5 face onnx推理,支持原版yolov5 face 导出的onnx,亲测成功
支持yolov5 face m模型,也支持yolov5 blazeface 。
yolov7 分割 tensorrt 推理源代码
yolov7 mask分割 tensorrt 推理源代码,网上没有人实现,和大佬一起熬了好几个通宵研发出来,支持tensorrt8,win10平台,亲测成功,可运行
linux也支持,可自行跨平台适配。
labelme-5.1.1
labelme-5.1.1,设置了自动保存,标签不保存图片,自定义标签颜色,亲测成功,
博客地址:
https://blog.csdn.net/jacke121/article/details/112288397
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,人脸检测器。
yolov7训练 json标注格式代码
yolov7训练只支持txt格式,自己添加了json标注格式解析和支持。
yolov7_mask_ncnn win10版
yolov7 mask ncnn c++ win10版,亲测成功。
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
yolov5 openvino2022版本
yolov5 openvino 2022版本,win10系统vs2019,摄像头实时预览版本,亲测调试成功,特此给广大学友们分享
yolov5 openvino版本
yolov5 openvino 2021版本,win10系统vs2019,摄像头实时预览版本,亲测调试成功,特此给广大学友们分享
yolov5-face c++ mnn版
yolov5-face c++ mnn版,亲测成功,
3399设备上 100ms 320*240
mnn windows centerface 示例项目
mnn center windows示例代码,包含mnn库,和centerface代码。
yolov5各个版本 tensorrt cfg文件
yolov5各个版本 tensorrt cfg文件,下载地址:
https://github.com/enazoe/yolo-tensorrt/blob/master/yolov5_tutorial.md
担心作者哪天不再共享,特地下载下来,分享给大家。
car_select.rar
coco2017筛选提取车辆脚本,并转为xml格式,并手工精选车辆样本数据,
去除严重遮挡,暴露面积超级小的等等其他极难样本。
coco2017转xml格式,精选人体数据集
coco2017转xml格式代码,并且手工挑选,去掉超级难度样本,留下精选样本列表
比如只有一只手,一只脚的样本;
剔除远处的超级样本,人体像素小于20个像素的超级小人体。
本人手工标注,花了3个周的时间,原创不易,请珍惜。
ellipse_detector.rar
windows c++ opencv 圆检测,椭圆检测,源码,亲测可行
Yolov5_DeepSort_bytetrack
Yolov5 DeepSort跟踪 bytetrack跟踪实例代码,
deepsort 跟踪示例代码:track.py
ByteTrack超越了此前所有的跟踪方法。
参考资料:https://blog.csdn.net/zhouchen1998/article/details/120932206
bytetrack示例代码:track_byte_demo.py
yolov5 代码版 芒果检测 带训练权重,可以直接用
yolov5 代码版 芒果检测
关于代码版,不是配置文件版,修改网络非常方便,参见我的博客:
https://blog.csdn.net/jacke121/article/details/118392154
芒果检测 带训练权重,可以直接用
旋转目标检测标签转换为yolov5格式
旋转目标检测标签转换,支持以下数据格式转换为yolov5 txt 模式:
dota 数据格式转换,
itc_vd数据格式转换
uavrod_xml 数据格式转换
ucas数据格式转换
训练代码:
https://github.com/BossZard/rotation-yolov5
ByteTrack.rar
最强行人检测跟踪ByteTrack,原版代码有bug,输入视频,会出现把视频清空的bug,此为修正版,效果比yolov5强很多
yolo系列 keams聚类anchors
yolo系列 keams聚类anchors
yolov5_trackid.rar
yolo5 目标检测,行人、车辆等物体检测、跟踪、计数,车辆方向检测
open3d_release_demo.rar
open3d 0.13最新版 win10 x64 release动态库,以及调用例子,亲测可用。
mnn_yolofastest.rar
yolofastest mnn版,mnn1.2.0,win10系统,vs2019,亲测可用。
yolov5-face_align.rar
yolov5-face 对齐人脸,yolov5-face 人脸检测,检测5个关键点,并对齐转正。
入库:detect_align.py
mnn windows 转换工具 1.2.0
mnn windows 转换工具 1.2.0,release版,目前最新版,编译了2个小时,亲测可用,现在来分享给大家,可以直接用
mnn1.2.0.zip
windows mnn x64位库 dll,lib文件,已经编译好了,直接用就行。
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
yolov5代码版,带预训练,测试脚本,可训练
yolov5配置版,阅读修改起来不太方便,特地整理出来代码封装版,阅读,修改网络比较方便,带预训练权重,检测代码:detect_class_s.py,检测芒果的,可以训练别的。