2021年01月_AI算法网奇

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

原创 python下载mp4

urllib.request.urlretrieve(video_url,'./videos/%s.mp4'%(title))下载多了不能用了； req = requests.get(url, headers=self.headers, stream=True, verify=False,timeout=500) with(open(target_dir + movie_name + '.mp4', 'wb')) as f: ...

2021-01-31 22:31:31 2626 1

原创 numba示例

# -*- coding: utf-8 -*-import osimport numbaimport numpy as npimport cv2from numba import jit@jitdef aaa(ccc): py_list = [0,1] number_list = numba.typed.List([0]) cnts = numba.typed.Dict.empty( key_type=numba.int64, valu...

2021-01-31 03:37:55 2308

原创 conda 更换python版本更改虚拟环境路径

如将Anaconda 中默认版本Python3.7 版本修改成3.5 执行下列命令即可conda install python=3.7.4自动会卸载之前的版本，更新新的版本。感谢：https://blog.csdn.net/qq_31904559/article/details/84309756

2021-01-23 18:30:01 30177 2

转载 DLL load failed while importing _pywrap_tensorflow_internal

前往微软官网下载Visual Studio C++即可：https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads

2021-01-23 18:11:14 3073

转载 mnn推理示例

MNN项目组已经提供了编译好的库：1.下载编译好的MNN库下载地址为：https://github.com/alibaba/MNN/releases下载两个文件：Source.code和MNN-WindowsX64-0.2.1.7.zip2.在visual studio上进行配置注意visual studio版本为2017，我习惯把这些库都跟opencv放一起，具体来讲就是：首先，把从Source code中解压的include文件如图1所示，复制出来；然后，在自己...

2021-01-23 15:04:14 3417

原创 linux 常用命令总结

创建目录：1 mkdir，创建一个空目录。比如：mkdir test2 mkdir -p，递归创建目录。比如：mkdir -p a/b。迭代创建。3 mkdir -m 777 d创建权限为777的目录4 命令参数。-m mode=模式，设定权限（类似chmod）。-p parents 递归创建目录。-v verbose 每次创建目录都显示信息。.....................

2021-01-22 18:24:09 2492 1

原创 pandas打乱数据

pandas自己的：shuffle_data=df.sample(frac=1).reset_index(drop=True)sklearn的：import numpy as npimport pandas as pdfrom sklearn.utils import shuffle as resetdef train_test_split(data, test_size=0.3, shuffle=True, random_state=None): '''Split ..

2021-01-22 14:16:46 2479 1

原创 python操作txt 写入列表

data=["asdf\n","34353\n"] file_w = open(r'stat.txt',"w") file_w.writelines(data) file_w.flush() file_w.close()

2021-01-21 21:31:06 4702 1

原创 python numpy转字符串

arr = np.array([ -1.8264365 , -0.6390996 , 5.1984577])print('arr',arr)# 数组arr转字符串strstr = ','.join(str(i) for i in arr)print('str:',str)

2021-01-21 21:27:21 8218

原创 repvgg

import torch.nn as nnimport numpy as npimport torchdef conv_bn(in_channels, out_channels, kernel_size, stride, padding, groups=1): result = nn.Sequential() result.add_module('conv', nn.Conv2d(in_channels=in_channels, out_channels=out_channe...

2021-01-21 19:29:31 2658

转载大改ShuffleNetV2网络，注意力机制，csp，卷积裁剪

大改ShuffleNetV2网络，注意力机制，csp，卷积裁剪1.背景在移动端场景中，目前有很多不错的轻量级网络可以选择，例如google的mobilenet系列，efficient lite系列，旷世的shufflenet系列，华为的ghostnet等。在这些模型中，我尤其喜欢shufflenetV2，因为它结构清晰，简洁（如下图所示为shufflenetv2 1.0x的结构图：红色标注为各个模块的flops），并且在设计之初就考虑了MAC（访存代价），使得在移动端部署的时候具有很低的延时。在多个

2021-01-19 22:49:10 5007 3

原创不允许dllimport函数的定义

不允许dllimport函数的定义为了减少使用dll时候的设置或者代码编写，把dll.h修改一下：#ifndef DLL_EXPORT#define DECLDIR __declspec(dllimport)#else#define DECLDIR __declspec(dllexport)#endifcpp中：解决方法：先定义宏：//#include "stdafx.h"#include <iostream>#define DLL_EXPORT //先...

2021-01-19 18:52:22 8523

原创 Parse error. Expected a command name

Parse error. Expected a command nameCMake Error at CMakeLists.txt:9:Parse error. Expected a command name, got unquoted argument with text"set（OpenCV_DIR".原因：出现了中文括号，正确代码：set (OpenCV_DIR E:/opencv/opencv)#opencvfind_package(OpenCV REQUIR...

2021-01-19 14:31:29 3869

原创 cmake找不到opencv解决

报错代码：find_package(OpenCV REQUIRED)解决方法1：主要问题在于库路径问题，应该在CMakeLists.txx中设置opencv路径指向build文件夹。 set（OpenCV_DIR /path/to/opencv/build） find_package(OpenCV REQUIRED) 方法2，ok：配一个环境变量：OPENCV_DIR 到bin目录的父目录 set(OPENCV_DIR_HINT "") ...

2021-01-19 14:21:33 7572

原创 gpu实时显示 nvidia-smi实时刷新

import time import subprocess import locale import codecs import os cmd = "nvidia-smi" interval = 3 while True: ps = subprocess.Popen(cmd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, shell=True) fo...

2021-01-18 23:26:35 9257

原创 python 点计算角度

# -*- coding: UTF-8 -*-import cv2import numpy as npimport osimg = np.zeros([512, 512, 3])pts = np.array([[50, 190], [380, 420], [255, 50], [120, 420], [450, 190]])# 第三个参数指的是是否封口,这里注意第二个参数外面必须再加一层中括号cv2.polylines(img, [pts], True, (255, 255, 0)...

2021-01-15 16:40:04 7768

原创关键点网络

https://github.com/3d-hand-shape/hand-graph-cnn# -*- coding: utf-8 -*-from __future__ import absolute_importfrom __future__ import divisionfrom __future__ import print_functionfrom __future__ import unicode_literalsimport osimport timei...

2021-01-13 22:06:50 2320

原创 python线段插值

# -*- coding: UTF-8 -*-import cv2import numpy as npimport osimport numpy as npimport matplotlib.pyplot as pltxn = [10,80,80,10]yn = [20,25,50,51]data=[]x_data=[]y_data=[]for index, x_v in enumerate(xn): next_index=index+1 if ind...

2021-01-13 12:20:17 2300

原创 python 遍历删除方法总结

python 遍历删除方法总结

2021-01-12 23:18:29 2869

原创 python 文件大小，获取时间

1.os.path.getsize(file_path)：file_path为文件路径import osos.path.getsize('d:/svn/bin/SciLexer.dll')#14799042. 遍历文件夹，将所有文件大小加和， os.walk()遍历文件夹import osfrom os.path import join, getsizedef getdirsize(dir): size = 0 for root, dirs, fil...

2021-01-11 23:32:18 2282

原创 ‘utf-8‘ codec can‘t encode character ‘\udcc0‘ in position 35

'utf-8' codec can't encode character '\udcd5' in position 64'utf-8' codec can't encode character '\udcc0' in position 35在网上看了一些基本上都是说文件的路径的问题，不排除是因为文件路径的问题。windows下文件夹的路径使用“\\”隔开，Linux中使用“/”，我简单的理解为windows下需要转义一下，不能直接使用“/”；如果在windows中使用pyth...

2021-01-11 22:41:04 3645

原创 linux pip 找不到python

linux系统下， pip 文件是个脚本文本，里面第一行写的python的路径conda 换磁盘地址后，pip 脚本中需要更新python的路径

2021-01-11 18:55:56 2522

原创 cuda 0 not supported xla

cuda 0 not supported xla这样貌似能解决：os.environ["CUDA_VISIBLE_DEVICES"] = "1"

2021-01-10 15:56:38 3528

原创多人脸跟踪

https://github.com/Linzaer/Face-Track-Detect-ExtractPython 3.5+ Tensorflow MTCNN Scikit-learn Filterpy

2021-01-09 22:57:05 2389

原创 python 点的投影变换

# -*- coding: utf-8 -*-import osimport numpy as npimport cv2# -*- coding:utf-8 -*-import cv2import numpy as npdef rad(x): return x * np.pi / 180img = np.zeros((400,400,3),dtype=np.uint8)# cv2.imshow("original", img)img[100,10]=(255...

2021-01-08 23:40:58 2589

原创 torch.load No module named

model = torch.load(test_pkl, map_location=lambda storage, loc: storage)报错：No module named 'networks.MSBDN-DFF-v1-1'原因：networks包下面没有MSBDN-DFF-v1-1 这个文件原因不知道为什么，解决方法，networks下面需要有MSBDN-DFF-v1-1.py。...

2021-01-07 23:21:21 3100 1

原创图像去躁2021

https://github.com/m-tassano/fastdvdnetpython的参考：https://blog.csdn.net/jacke121/article/details/79945538

2021-01-07 20:34:37 4019

原创去雪网络

https://github.com/weitingchen83/JSTASR-DesnowNet-ECCV-2020

2021-01-07 18:26:23 2430

原创 pytorch MaxUnpool2d

这个函数onnx不支持Upsampling:上采样，被nn.functional.interpolate代替 nn.Upsampling is deprecated. Use nn.functional.interpolate instead.pytorch: MaxUnpool2d 与 Upsampling#-*-coding:utf-8-*-import torchimport torch.nn as nnimport numpy as np ...

2021-01-07 15:38:29 3450

原创实时语义分割总结

实时语义分割总结

2021-01-07 14:03:07 2718

转载 ace 图像增强

# -*- coding: utf-8 -*-import osimport numpy as npimport cv2import cv2import numpy as npimport mathdef stretchImage(data, s=0.005, bins=2000): # 线性拉伸，去掉最大最小0.5%的像素值，然后线性拉伸至[0,1] ht = np.histogram(data, bins); d = np.cumsum(ht[0]) / f...

2021-01-07 10:01:21 2576 1

原创 labelme标注工具安装改进

线的粗细：0是很细，1中等 ) pen = QtGui.QPen(color) # Try using integer sizes for smoother drawing(?) pen_width=max(1, int(round(1.0 / self.scale))) pen.setWidth(0) painter.setPen(pen) lin...

2021-01-06 18:08:17 3081 1

原创 python 计算斜率

p1=[1,2] p2=[3,4] xielv=abs((p1[1]-p2[1])/( p1[0]-p2[0]+1e-5)) if xielv>0.25 and xielv<2: print(xielv)

2021-01-05 10:50:31 8017

原创 Unexpected key(s) in state_dict: “dense_block1.denselayer1.norm.1

Unexpected key(s) in state_dict: "dense_block1.denselayer1.norm.1from torchvision.models import densenet121from collections import OrderedDictmodel = densenet121(pretrained=False)state_dict =torch.load(model_weight_path)# 初始化一个空 dictnew_sta...

2021-01-04 00:08:08 2353 2

原创去雾算法2020

实验选择去雾算法研究中常用的有雾图像进行测试实验1：现有经典去雾算法:MSCNN算法、DehazeNet算法、暗通道先验算法进行比较

2021-01-03 22:16:25 6295

原创 torch.utils.ffi is deprecated. Please use cpp extensions instead

torch.utils.ffi is deprecated. Please use cpp extensions insteadtorch1.6第一步在build.py文件中的找到语句：from torch.utils.ffi import create_extension(共一处）将其改为：from torch.utils.cpp_extension import BuildExtension第二步在build.py文件中的找到语句ffi = create_extension...

2021-01-03 16:24:51 3712 4

原创 Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd

Legacy autograd function with non-static forward method is deprecated. Please use new-style autograd function with static forward method jacke121报错代码：self.priors_pal2.type(type(x.data)).to(self.device)解决方法：

2021-01-01 15:39:58 4234

转场分割代码封装源代码

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