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 2596 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 2263

原创 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 28665 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 2965

转载 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 3368

原创 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 2268

原创 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 2424 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 4595

原创 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 8140

原创 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 2632

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

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

2021-01-19 22:49:10 4825 2

原创不允许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 7951

原创 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 3678

原创 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 7023

原创 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 8987

原创 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 7395

原创关键点网络

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 2291

原创 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 2281

原创 python 遍历删除方法总结

python 遍历删除方法总结

2021-01-12 23:18:29 2836

原创 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 2260

原创 ‘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 3448

原创 linux pip 找不到python

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

2021-01-11 18:55:56 2495

原创 cuda 0 not supported xla

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

2021-01-10 15:56:38 3464

原创多人脸跟踪

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

2021-01-09 22:57:05 2358

原创 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 2566

原创 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 3042 1

原创图像去躁2021

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

2021-01-07 20:34:37 3885

原创去雪网络

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

2021-01-07 18:26:23 2376

原创 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 3411

原创实时语义分割总结

实时语义分割总结

2021-01-07 14:03:07 2679

转载 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 2552 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 2982 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 7850

原创 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 2324 2

原创去雾算法2020

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

2021-01-03 22:16:25 6100

原创 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 3659 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 4061

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关注的人

rknn 麒麟系统adb驱动安装

BIWI vocaset模板

rknn yolov5 rtsp

问财选股导出excel pythony源代码

自研lstm多目标跟踪，DanceTrack验证集精度98%

强大好用的人体关键点标注工具

pyopengl windows x64 whl安装包

mmpose 预测转json标注

3d-nms cuda代码

MobileViT v3 pytorch代码

OpenPCDet ingroup-inds库 windows编译

Ultra-Fast-Lane-Detection-v2 pytorch版

insightface/alignment 人脸关键点的pytorch版本

yolov5 face onnx推理代码

yolov7 分割 tensorrt 推理源代码

labelme-5.1.1

3DDFA-V2.5-opencv-yolov5face

yolov7训练 json标注格式代码

yolov7_mask_ncnn win10版

tensorrt c++ 推理项目

yolov5 openvino2022版本

yolov5 openvino版本

yolov5-face c++ mnn版

mnn windows centerface 示例项目

yolov5各个版本 tensorrt cfg文件

car_select.rar

coco2017转xml格式，精选人体数据集

ellipse_detector.rar

Yolov5_DeepSort_bytetrack

yolov5 代码版 芒果检测 带训练权重，可以直接用

旋转目标检测标签转换为yolov5格式

ByteTrack.rar

yolo系列 keams聚类anchors

yolov5_trackid.rar

open3d_release_demo.rar

mnn_yolofastest.rar

yolov5-face_align.rar

mnn windows 转换工具 1.2.0

mnn1.2.0.zip

256_256_resfcn256_weight.index

神经网络改进的几个思考

yolov5 代码版芒果检测带训练权重，可以直接用