2020年04月_AI算法网奇

12月 11月 10月 09月 08月 07月 06月 05月 04月 03月 02月 01月

原创 pytorch维度统计

统计第2维数据全部小于3的 import torch data_a=torch.arange(0, 6) data= data_a.reshape([3,2]) data_b= data_a.clone().reshape([3,2]) data[data<3]=0 print(data) data2=torch.sum(...

2020-04-30 23:42:37 2451

原创可分离卷积

https://baijiahao.baidu.com/s?id=1634399239921135758&wfr=spider&for=pc第1部分-深度卷积:第2部分-逐点卷积:

2020-04-30 11:52:44 3579

原创神经网络分类支持子文件夹

def is_number(s): try: float(s) return True except ValueError: pass try: import unicodedata unicodedata.numeric(s) return True exce...

2020-04-30 11:14:25 2498

原创 3D人脸重建

地址在https://github.com/YadiraF/PRNet

2020-04-29 18:35:48 2587

原创 android camera之nv21旋转

android camera之nv21旋转这周做的一个android的camera开发，需要获取到视频帧数据，并且需要是nv21格式的byte数组，并且视频帧的图像需要是正方向的。和android相机打过交道的都清楚，android的camera获取到的图片都是横向的，因此，需要进行旋转，对于图像的旋转，其实bitmap这个类已经可以帮我们实现了，但是前提是你需要将你的数据格式转换为Bi...

2020-04-29 00:28:56 3257

原创 c++调用python返回字典

C++调Python，代码粘贴如下: 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 ...

2020-04-29 00:28:24 4415

原创 PyObject_CallMethod self问题

类函数的第1个参数是self，c++传参没有传self，导致python接收参数错误void main() { Py_Initialize(); PyObject* module = PyImport_ImportModule("bbb"); assert(module != NULL); // Returned non-null object PyObject* MyPyC...

2020-04-28 17:29:37 4055 2

原创 PyObject_CallMethod

PyObject_CallMethod(myobject, "dosomething", "s", "blahdy blah");

2020-04-28 16:47:08 6474

原创 PyImport_ImportModule返回值=null

PyImport_ImportModule返回值=null原因：1. 添加路径PyRun_SimpleString("import sys"); PyRun_SimpleString("sys.path.append('./')");2.py文件中有错误3.

2020-04-28 16:31:20 3756 7

原创 char与wchar_t(TCHAR)之间的相互转换

char与wchar_t(TCHAR)之间的相互转换1、wchar_t类型数据的初始化char类型数据的初始化就不再骜述了；只说一下wchar_t类型的初始化：wchar_t[] = L"我是wchar_t类型的数据"; //wchar_t类型数据初始化时前面需加上宏L2、char与wchar_t之间的转换char与wchar_t类型数据之间的转换不能通过赋值的方...

2020-04-28 16:22:45 2708

原创 c++调用python3

参考：https://blog.csdn.net/love_clc/article/details/76653100c++访问python3-实例化类的方法原创孟思明最后发布于2017-08-04 08:34:44 阅读数 3787 收藏展开此文是学习笔记，供日后翻阅。下面列出C++访问python所需的函数，按调用的先后次序排列。提示：其中实例化类的方法...

2020-04-28 15:14:01 2489

原创 c++ 调用python2类获取返回值

前面介绍了用python调用C++时用swig工具，现在简单介绍一下C++调用python并取返回值的过程python文件test.pyclass tester: def add(self, a, b): return a + b新建一个VS控制台项目main.cpp所有的源码如下:#include <stdio.h>#include ...

2020-04-28 14:59:06 2799

原创 pycharm 转大写

pycharm 转大写默认是选中 + ctrl/command + shift + u

2020-04-28 14:38:31 3862

原创 c++字符串

char greeting[] = "Hello";char* tmp ="ABCDEF";

2020-04-28 12:29:58 2637

原创 anynet

# -*- encoding: utf-8 -*-"""@File : AnyNet.py@Time : 2020/4/13 14:39@Author : Alessia K@Email : ------"""import timeimport torchimport torch.nn as nn# from utils.parse_cfg impo...

2020-04-28 11:19:01 3113

原创 module ‘yaml‘ has no attribute ‘FullLoader‘

module 'yaml' has no attribute 'FullLoader'

2020-04-28 10:05:42 5378

原创 AssertionError: SRE module mismatch

AssertionError: SRE module mismatchTraceback (most recent call last): File "D:/project/pose/simple-HRNet/models/hrnet.py", line 1, in <module> import torch File "D:\ProgramData\M...

2020-04-27 23:34:43 6920 1

原创 hrnet笔记

中科大和微软亚洲研究院，发布了新的人体姿态估计模型，刷新了三项COCO纪录，还中选了CVPR 2019。这个名叫HRNet的神经网络，拥有与众不同的并联结构，可以随时保持高分辨率表征，不只靠从低分辨率表征里，恢复高分辨率表征。如此一来，姿势识别的效果明显提升：在COCO数据集的关键点检测、姿态估计、多人姿态估计这三项任务里，HRNet都超越了所有前辈。...

2020-04-27 22:23:28 2729

原创车牌识别学习资料整理

https://github.com/xuexingyu24/License_Plate_Detection_Pytorch

2020-04-27 12:54:19 2150 2

原创 one of the variables needed for gradient computation has been modified by an inplace

one of the variables needed for gradient computation has been modified by an inplace参考：https://blog.csdn.net/qq_41997920/article/details/100095961由于新版本的pytorch把Varible和Tensor融合为一个Tensor，i...

2020-04-26 21:48:11 4453

原创 python编码格式

编码格式：# -*- coding: utf-8 -*-这个是文件头，还有编码是文件的编码格式。

2020-04-26 18:09:33 2428

原创 tx2 pytorch

1.sudo apt-get install python3-pip2.sudo apt install liblapacke-dev checkinstall # For OpenCV3.sudo apt-get install libopenblas-dev4.sudo pip3 install cython5.sudo pip3 install numpy scipy...

2020-04-26 14:23:05 2392

原创 ImportError: libopenblas.so.0: cannot open shared object file

ImportError: libopenblas.so.0: cannot open shared object file解决：sudo apt-get install libopenblas-dev

2020-04-26 14:11:19 10443

原创 hardnet68尝试

接ssd的，收敛比较慢import osimport torchimport torch.nn as nnimport torch.nn.functional as Fclass Identity(nn.Module): def __init__(self): super(Identity, self).__init__() def forwar...

2020-04-26 12:45:57 2819

原创 rfbnet笔记

很感谢这个作者，写的很好https://zhuanlan.zhihu.com/p/76491446主要追求GPU的速度更快，因此也没有去使用分组卷积深度可分离卷积，后面会尝试使用不同的网络设计。整体上与RFB300类似，不过有以下几点不同：削减VGG的通道数目以及卷积数目，通道数目最多也只有128原始VGG没有BN，这里加上BN，梯度更加稳定，学习率也可以加大些使用1x...

2020-04-26 00:29:05 2456

原创 yolov4

c++版 qthttps://github.com/scutlrr/yolov4_QtGUI

2020-04-25 21:40:25 5040

原创正则化方法之DropBlock

参考：https://blog.csdn.net/qq_14845119/article/details/85103503正则化方法之DropBlock论文：DropBlock: A regularization method forconvolutional networksGithub：https://github.com/miguelvr/dropblockhttp...

2020-04-24 23:43:46 2666

原创 C++ 32位和64位

C++32位和64位程序的区别原文地址：https://blog.csdn.net/nma_123456/article/details/45077345由于操作系统内存分配的不同，导致软件开发过程中，需要编译不同版本的软件，此处以VS为例。1、编译程序根据需要选择不同的编译环境：X86和win32位32位程序，X64为64位程序，可以选择不同的编辑条件形成不同位的...

2020-04-24 20:43:03 3563

原创分类预测笔记

# -*- coding: utf-8 -*-from __future__ import print_function, divisionimport argparseimport osimport cv2import torchimport torch.nn.functional as m_funcfrom torchvision import transforms...

2020-04-24 11:25:53 1308

原创 ALFNet行人检测

实际测试，发现检测完全错误。可能是官方开源的model有问题，也可能是参数设置的问题(见Config.py文件)地址：https://github.com/liuwei16/ALFNet2年前的：https://github.com/VideoObjectSearch/ALFNet也已经证明：通过在Faster R-CNN的第二阶段级联多步预测会进一步提升Faster ...

2020-04-24 00:09:58 2971

原创 vgg16测试

1070 640 b:4 3ms多了就内存溢出import timeimport torchfrom torch import nnclass S3FDNet(nn.Module): def __init__(self, device='cuda'): super(S3FDNet, self).__init__() self.vg...

2020-04-23 18:43:28 2732

原创 python32位安装

安装32位查看32位：cmd打开命令行，输入python，查看。如果32bit，则是32位；如果是64，则是64位

2020-04-23 12:16:39 3179

原创 CUDA out of memory in pytorch

with torch.no_grad():CUDA out of memory in pytorch今天尝试了一下Transformer，一直遇到当validate若干次之后爆显存，一开始以为参数过多，batch size过大，但是无济于事。谷歌了相关内容，发现原来是需要加with torch.no_grad():...

2020-04-22 13:08:00 2940 1

原创 PyramidBox笔记

PyramidBoxhttps://github.com/aspxcor/Pyramidbox-Based-Head-Recognition/tree/master/PyramidBox-master首先简单介绍了人脸检测的发展，SSH、S3FD通过设计具有尺度不变性的网络结构，在单个网络的各个特征金字塔上检测不同尺度的人脸。如何通过人脸的上下文信息（contextual in...

2020-04-21 23:15:29 2499

原创 The NVIDIA driver on your system is too old

The NVIDIA driver on your system is too oldhttps://www.zhihu.com/question/309583980原因是我安装的pytorch版本是需要cudav9.2版的，而我电脑上安装的是v9.0版的，不兼容因此，解决方案是安装cudav9.2版，问题就解决了0人点赞pytho...

2020-04-21 17:01:26 4645

原创 cannot import name '_imaging' from 'PIL'

cannot import name '_imaging' from 'PIL'解决方法：先卸载掉原版本的pillow：pip uninstall pillow在安装4.0.0版本的pillow:pip install pillow==4.0.0

2020-04-21 16:25:26 13055 7

原创 onnx c++

opencv:https://github.com/tw0226/onnx_model_to_C-/blob/bb54aef3db3b9f37229c8fbc27d2b64db6950924/onnx_convert/main.cpp

2020-04-20 18:10:31 5222

原创风格迁移开源推荐

c++ onnxhttps://github.com/NeuralVFX/style-transfer-open-cv

2020-04-20 15:14:57 2494

原创 IResNet

代码链接：https://github.com/iduta/iresnetIResNet参考：https://mp.weixin.qq.com/s?__biz=MzUxNjcxMjQxNg==&mid=2247498263&idx=2&sn=6bf173ad89b9af457a64674be166bcc7&chksm=f9a18898ced601...

2020-04-20 14:46:11 4038 2

本算法根据TransNetV2-SBD 做了代码封装，可以直接用来进行转场判断，也支持直接把mp4视频根据转场分割平台是基于TensorFlow的，执行入口transnetv2_demo.py TensorFlow gpu和cpu都支持： tensorflow 2.12.0 tensorflow-estimator 2.12.0 tensorflow-gpu 2.8.0 tensorflow-intel 2.12.0 tensorflow-io-gcs-filesystem 0.31.0 termcolor 2.3.0

2024-09-17

opencv 中文字体库

2024-09-17

自研人脸关键点，轻量级高精度人脸关键点

自研人脸关键点，轻量级高精度人脸关键点98个点，模型大小700多kb 低头，抬头，侧脸，都可准确识别，可以转ncnn，mnn在手机上cpu到3毫秒以内带人脸跟踪算法；包含训练代码： train_lbg_box_new.py 推理代码 demo_v5_face.py 人脸检测、人脸关键点综合示例代码； demo_v5_nopad_last.py yolov5 face onnx转ncnn代码： Peppa-Facial-Landmark-kuoda\deploy\onnx_2_ncnn_yolov5.py 人脸关键点onnx转ncnn： onnx_2_ncnn_mark.py

2024-08-15

yolov10最新源码版，训练支持labelme标注

yolov10最新源码版，原版只支持标注格式为txt格式，笔者以前训练yolo用labelme的json格式标注，统一格式，数据管理，可视化，查badcase更方便。修改了数据格式，增加支持labelmejson独立格式标注标注样例： { "version": "5.3.1", "flags": {}, "shapes": [ { "label": "dao", "points": [ [ 1990.9259259259259, 322.3395061728395 ], [ 53.75, 689.5000000000001 ] 预测发现还是会有重叠框，加上nms过滤重叠框，效果会更好本预测代码已经加上nms机制

2024-06-14

pyqt opengl 3d渲染源码正交投影，透视投影

pyqt opengl 3d渲染源码支持格式，3d画线，立方体（也可以进行扩展二次开发），可以旋转，平移，缩放支持鼠标点选提示，鼠标和立方体交点3d坐标，代码经过反复优化，测试，效果比较稳定。最新代码： widget_opengl_change1.py 通过圆柱组合形成立方体，支持正交投影，透视投影无缝切换，左下角显示3d坐标。其他代码也有学习和参考价值

2024-06-12

rk3588 麒麟系统硬解码硬编码源代码

rk3588 麒麟系统硬解码h264 硬编码264源代码 qt c++ 项目，亲测成功项目带有librockchip-mpp安装文件， librockchip-mpp-dev_1.5.0-1rk4_arm64.deb 安装rockchip-mpp deb sudo dpkg -i librockchip-mpp-dev_1.5.0-1rk4_arm64.deb 使用终端命令和apt工具：打开终端，使用以下命令安装deb文件并解决依赖关系： sudo apt install ./librockchip-mpp-dev_1.5.0-1rk4_arm64.deb 需要提前安装ffmpeg库，main.cpp中有回调函数，mpp_decoder_frame_callback，用来接收解码后的frame，经过测试，效率挺高的，可以达到30帧以内。

2024-06-08

qt c++ http-server 多线程源代码

qt c++ 实现 http_server 在子线程启动，收到消息后，更新主线程的ui显示，不卡顿本人亲测成功，希望给大家提供便利。跨平台的，Windows，linux，arm，麒麟系统都能用

2024-06-07

网络流量异常检测系统源代码

网络流量异常检测系统基于Network-anomaly-detection-with-deep-learning-along-with-UI 做的改进改进1 抓包工具更新 Npcap下载： https://npcap.com/#download 改进2，原来代码用了多线程，在Windows系统运行会导致死机，本次针对死机做了改进改进2，源代码对pandas 的dataframe操作导致频繁报警，改进之后频繁报警没有了。

2024-06-01

麒麟3588系统语音播报源代码

麒麟3588系统语音播报源代码麒麟arm系统qt c++文本生成播报系统源代码可以定制播报人性别，多款播报人可供选择，支持中英文，c++ qt示例源代码，亲测成功。 QListWidget 自定义样式，解决QPixmap不能读取图片的问题。

2024-05-30

qt c++ sqlite 封装

2024-05-27

自研扩散模型高光谱修复网络

自研扩散模型高光谱修复网络基于MST_Plus_Plus 网络改造。试验数据扩散模型loss初步测试降到了0.005，比不加扩散loss小了20倍，训练入口 train_cos_img.py

2024-05-18

bvh最新播放器 python源代码

bvh最新播放器可以显示骨骼名称，旋转角度，自适应大小，支持3维npz数据可视化 python实现，提供源代码，修改和完善很方便。根据3维npz生成bvh bvh最新播放器 python源代码详细介绍： https://blog.csdn.net/jacke121/article/details/138809268

2024-05-13

rknn 麒麟系统adb驱动安装

2024-04-28

BIWI vocaset模板

BIWI vocaset模板包括FLAME_sample.ply FLAME_masks.pkl BIWI/templates.pkl BIWI/F1.obj BIWI.ply

2024-04-25

rknn 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

yolov7训练 json标注格式代码

yolov7训练只支持txt格式，自己添加了json标注格式解析和支持。

2022-10-06

yolov7_mask_ncnn win10版

yolov7 mask ncnn c++ win10版，亲测成功。

2022-09-30

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

神经网络改进的几个思考

2022-04-09

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

转场分割代码封装源代码

opencv 中文 字体库