2020年03月_AI算法网奇

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

原创 pytorch 加载模型：

1.直接加载网络import torchpthfile = r'E:\anaconda\app\envs\luo\Lib\site-packages\torchvision\models\squeezenet1_1.pth'net = torch.load(pthfile)print(net)

2020-03-31 19:14:28 2633

原创 [TensorRT] ERROR: Network must have at least one output

[TensorRT] ERROR: Network must have at least one output

2020-03-31 18:59:03 6775

原创 cannot import name &#39get_all_providers&#39

cannot import name 'get_all_providers'C:\Users\ADMINI~1\AppData\Local\Temp\2\_MEI87882\onnxruntime\capi\_pybind_state.py:13: UserWarning: Cannot load onnxruntime.capi. Error: 'DLL load failed: 找不...

2020-03-31 16:22:27 5967

原创 cannot import name 'get_all_providers'

2020-03-31 16:20:10 4557 3

原创 pytorch判断NaN

pytorch判断NaNYou can always leverage the fact thatnan != nan:data = torch.tensor([1, 2, np.nan])tensor([ 1., 2., nan.])data[data != data]tensor([ 0, 0, 1], dtype=torch.uint8)Wi...

2020-03-31 11:38:20 9391 2

原创 jetpack 安装

查看jetpack的版本head -n 1 /etc/nv_tegra_release查看TensorRT的版本：dpkg -l | grep TensorRTjetpack安装教程：https://www.jianshu.com/p/e5bc3dc9453c

2020-03-30 14:43:25 3126

原创 windows tensorrt python

https://github.com/faedtodd/Tensorrt-yolov3-win10

2020-03-29 16:23:19 4527 3

原创 Pytorch TensorRT

https://blog.csdn.net/qq_38003892/article/details/89314108Pytorch通过保存为ONNX模型转TensorRT5原创小关学长最后发布于2019-04-15 16:08:35 阅读数 5506 收藏展开1 Pytorch以ONNX方式保存模型 def saveONNX(model, filepath):...

2020-03-29 15:55:44 3364

原创 Windows 软件推荐2020

Windows可谓是大多数人的生产力工具，集娱乐办公于一体，虽然在程序员这个群体中都说苹果是信仰，但是大部分不都是从Windows过来的，而且现在依然有很多的程序员用Windows。所以，今天我就把我私藏的Windows必装的软件分享给大家，如果有一个你没有用过甚至没有听过，那你就赚了????，这可都是提升你幸福感的高效率生产力工具哦！NO、1 ScreenToGif屏幕，摄像头和...

2020-03-27 17:22:12 2708

原创 PyTorch 按索引赋值

PyTorchif __name__ == '__main__': import torch x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], dtype=torch.float) index = torch.tensor([0, 1,2]) x.index_fill_(1, inde...

2020-03-27 17:20:53 6829

原创图像分割2021

https://github.com/WXinlong/SOLO今天跟大家分享下前天新出的论文SOLOv2: Dynamic, Faster and Stronger，一看这名字就知道很霸气！SOLO 超越自己，在原有架构基础上引入动态卷积和发明了Matrix NMS，再次刷新COCO 实例分割数据集的最高精度（41.7 AP），而且其轻量级版本精度与Mask RCNN相近...

2020-03-27 00:08:48 2984

原创 retinaface训练笔记

注意：73上map连续几天一直低，因为负样本loss由0.01改为了0.02网络的瓶颈：先预测分类，然后在分类正确的做回归，wh的回归98%，但是xy的回归只有70多，所以瓶颈在xy的回归上。如果是因为anchors的原因，导致map下降，但是实际情况是分类还是比较准的，是xy的回归不对,而且是正样本的xy回归不对，召回率低。代办问题：2020.01.20 72上，把xy和w...

2020-03-26 12:11:52 2143

原创 33 代替55

参考：https://www.zhihu.com/question/265791259主要原因很简单，因为两个3x3的卷积核的感受野的大小刚好是5x5。看了高赞的还是没明白，自己想了一下。比如，一个原始矩阵是2*3，卷积核是3*3，那么卷积结果是4*5的矩阵一个4*5的矩阵继续和3*3的卷积核卷积，结果是6*7的矩阵2*3的直接和5*5的卷积，结果是6*7的矩阵。...

2020-03-26 00:29:50 3012

原创 33卷积+13卷积+3*1卷积=白给的精度提升

ICCV2019: 通过非对称卷积块增强CNN的核骨架下面要介绍的论文发于ICCV2019，题为「ACNet:Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric ConvolutionBlocks」，axriv地址为：https://arxiv.org/abs/1908.03930v1。由...

2020-03-26 00:19:59 9974 1

原创 call() got an unexpected keyword argument 'partition_info'

__call__() got an unexpected keyword argument 'partition_info'

2020-03-25 18:18:18 4195

原创图像文档变形恢复DewarpNet

图像文档变形恢复DewarpNet：基于3D和2D堆叠回归网络的图像文档变形恢复https://zhuanlan.zhihu.com/p/102459708

2020-03-24 22:47:38 4171

原创 CentripetalNet 48

CentripetalNet 48论文链接：https://arxiv.org/pdf/2003.09119.pdf代码将在https: // github.com/kiveedong/petalnet上提供

2020-03-24 22:13:16 3050

原创 python编译

pythonsetup.pybuild_ext --inplace编写setup.py文件，获取帮助：python setup.py --help-commands[python]Standard commands: build build everything needed to install build_py...

2020-03-24 18:58:25 2839

原创 Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False

Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False此时改为：torch.load("0.9472_0048.weights",map_location='cpu')

2020-03-24 18:10:15 8884

原创 you're probably running inside a thread without first calling pythoncom.CoInitialize

you're probably running inside a thread without first calling pythoncom.CoInitialize后来在网上搜索，看到必须添加初始化函数和去初始化函数import pythoncomimport wmipythoncom.CoInitialize ()c = wmi.WMI ()#...

2020-03-24 16:55:58 3196

原创 some of the strides of a given numpy array are negative

内存地址不连续的问题加一句np.ascontiguousarray就好了报错pytorch使用cv2读取图片，并转换成RGB顺序的时候，在用torch.from_numpy转换成tensor的时候，报错：{ValueError}some of the strides of a given numpy array are negative. This is currently...

2020-03-24 00:30:24 2803

原创 python 加密

# -*- coding: utf-8 -*-# @Time : 2020/2/17 15:01# @Author : zzw# @File : spider_encrypt.py# @Software: PyCharmimport datetimeimport osimport randomfrom binascii import b2a_hex, a2b...

2020-03-23 23:24:19 1493

Baidu VIS 新出品的火腿框人脸检测器(HAMBox: Delving into Online High-quality Anchors Mining for Detecting Outer Faces)，刷新了最新的SOTA，将WIDERFACE 验证集hard子集的AP刷新到恐怖的93.3%，且基于HAMBox获得了ICCV WIDER Challenge人脸检测比赛冠军。该论文被...

2020-03-23 22:56:16 2814

原创 python获取硬件信息

pip install wmipip install pywin32复制代码import wmic = wmi.WMI()# # 硬盘序列号for physical_disk in c.Win32_DiskDrive(): print(physical_disk.SerialNumber)# CPU序列号for cpu in c.Win32_Processor()...

2020-03-23 18:28:37 2639

原创 invalid ELF header

invalid ELF header报类似错误：/usr/lib/x86_64-linux-gnu/libc.so: invalid ELF header一般应该是编译出的库文件或者二进制文件和运行环境不匹配。注意使用 file 命令，查看编译出的库和二进制文件的信息。root@ubuntu:/neo/c# file libs/armeabi/libhello...

2020-03-23 11:21:58 14732

原创人脸对齐端到端Super-FAN

Super-FANhttps://blog.csdn.net/qq_34919953/article/details/83449435

2020-03-23 00:47:47 2541

原创 mobilefacenet

权重4mclass MobileFaceNet(Module): def __init__(self, embedding_size): super(MobileFaceNet, self).__init__() self.conv1 = Conv_block(3, 64, kernel=(3, 3), stride=(2, 2), padding=(...

2020-03-23 00:33:41 1757

原创不确定性人脸识别 Uncertainty-aware Face Representation and Recognition

Uncertainty-aware Face Representation and Recognitionhttps://github.com/seasonSH/Probabilistic-Face-EmbeddingsModel Method LFW IJB-A (FAR=0.1%) 64-CNN CASIA-WebFace Baseline...

2020-03-23 00:00:35 2923

原创 IR-Net 测试

IR-Nethttps://github.com/htqin/IR-Net640 gpu测试50多msimport timeimport torchimport torch.nn as nnimport mathimport torch.utils.model_zoo as model_zooimport syssys.path.append('./modules...

2020-03-22 18:06:49 3381

原创 retinaface 检查增强 data_augment

crop后，有增强，有放大，有缩小宽高不齐的话会做粘贴增强，不够img_size会做放大缩小增强标签除以宽高，变成小数，就解决了缩放问题。import cv2import numpy as npimport randomfrom utils.box_utils import matrix_iofdef _crop(image, boxes, labels, img_siz...

2020-03-21 20:08:11 121 1

原创 pytorch crossentropy为nan

pytorch crossentropy为nan交叉熵损失函数的具体为：loss = -(x*ln(z)+(1-x)*ln(1-z))z = softmax(pred_x)这样当z为0/1时会出现loss为nan的情况本人的具体原因网络中用了MultiHeadAttention,attention的mask全为0,这样attention就为nan，造成个别样本的输出特征全...

2020-03-21 17:25:50 5439 2

原创任意方向目标检测

权重400多mhttps://github.com/MingtaoFu/gliding_vertex

2020-03-20 22:46:34 3968

原创 jittor框架

sudo apt install python-devsudo pip install https://github.com/Jittor/jittor.gitpython3 -m jittor.test.test_example

2020-03-20 11:53:39 3511

原创 Can't get source for site-packages\torchvision\ops\poolers.py. TorchScript requires source access in

Can't get source for site-packages\torchvision\ops\poolers.py. TorchScript requires source access in order to carry out compilation. Make sure original

2020-03-20 11:33:48 4158 1

原创 pyinstaller 'utf-8' codec can't decode byte 0xce in position 123: invalid continuation byte

pyinstaller 'utf-8' codec can't decode byte 0xce in position 123: invalid continuation byte在你打包的命令行中先输入chcp 65001然后再输入打包命令。pyinstaller -F xxx.py...

2020-03-20 10:45:10 2864

原创 torch 列最大值

max_v= torch.max(scores_s, 1)[0]下面的代码是一样的： index_num = torch.arange(0, conf.size(0)) max_ids= torch.max(scores_s, 1)[1] scores_v=scores_s[index_num,max...

2020-03-20 00:28:13 4767

原创 python 编译 exe和库

python 编译 exe和库

2020-03-19 17:32:05 3028 1

原创 retina多分类

配置文件：train_config.ini[default]rate = 0.01save_weight = 0[train]lr = 0.01cls_beishu = 1beishu_pos = 0.1beishu_neg = 0.5neg_count = 2l_neg = 0l_pos = 2l_xy = 2l_wh = 3l_map_score = 3...

2020-03-19 17:30:00 123 1

转场分割代码封装源代码

本算法根据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关注的人