自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(69)
  • 资源 (78)
  • 问答 (1)
  • 收藏
  • 关注

原创 selayer笔记

先做平均赤化,再做线性分类,再relu,再线性分类我的理解是增强分类的。下面是torch代码:class SELayer(nn.Module): def __init__(self, channel, reduction=4): super(SELayer, self).__init__() self.avg_pool = nn.A...

2020-02-29 22:13:25 5138 1

原创 vgg16学习笔记

VGG中根据卷积核大小和卷积层数目的不同,可分为A,A-LRN,B,C,D,E共6个配置(ConvNet Configuration),其中以D,E两种配置较为常用,分别称为VGG16和VGG19上图中,每一列对应一种结构配置。例如,图中绿色部分即指明了VGG16所采用的结构。我们针对VGG16进行具体分析发现,VGG16共包含:13个卷积层(Convolutional...

2020-02-29 19:50:17 3079

原创 GhostNet 测试

1070 640*480 batch 8可以https://github.com/iamhankai/ghostnet.pytorch/blob/master/ghost_net.py"""Creates a GhostNet Model as defined in:GhostNet: More Features from Cheap Operations By Kai Han, Y...

2020-02-28 13:00:04 3518 1

原创 vovnet 测试

vovnet39, 1070 640*640 batch 1 15ms最多 batch 4 130msimport timeimport torchimport torch.nn as nnimport torch.nn.functional as Ffrom collections import OrderedDict__all__ = ['VoVNet', 'v...

2020-02-28 11:30:21 3149

原创 bagnet 测试

1060上,1个640*640都跑不了import timeimport torch.nn as nnimport mathimport torchfrom collections import OrderedDictfrom torch.utils import model_zooimport osdir_path = os.path.dirname(os.path....

2020-02-28 11:21:32 2685

原创 转TensorRT

https://github.com/NVIDIA-AI-IOT/torch2trt

2020-02-27 19:49:45 2576

原创 RuntimeWarning: invalid value encountered in true_divide

RuntimeWarning: invalid value encountered in true_divide这个问题可能是在使用numpy的时候出现了0除以0造成的。比如:import numpy as npa = np.array([0,0])a/0__main__:1: RuntimeWarning: invalid value encountered in true...

2020-02-27 18:25:48 5129

原创 PyCharm与git/GitHub关联 pip install GitPython

我们发现,将project上传至GitHub,本质是一个git版本控制问题:在进行git版本控制后,有些时候会产生麻烦比如你新建了一个new file,但如果你不把它添加到git中,它就会变为红色那么如何取消Pycharm与GitHub的关联呢?1. 简单方法:2. 彻底一点的方法:首先进入project的根目录中,右击鼠标打开Git B...

2020-02-26 23:10:10 3922 1

原创 ONNX+TensorRT

使用ONNX+TensorRT部署人脸检测和关键点250fpsThis article was original written by Jin Tian, welcome re-post, first come with https://jinfagang.github.io . but please keep this copyright info, thanks, any ques...

2020-02-26 13:04:04 12304 1

原创 onnx动态分辨率实战

onnx导出动态分辨率pytorch1.4.0版本,不挑cuda版本,mxnet tf挑cuda版本 output_onnx = 'abcd.onnx' input_names = ["images"] output_names = ["detects"] inputs = torch.randn(1, 3, 360, 640) dynamic_...

2020-02-26 09:04:34 8141 2

原创 onnx retina人脸检测

import osimport timefrom math import ceilimport onnxruntimeimport numpy as npimport cv2import argparseimport argparseimport numpy as npfrom data import cfg_mnet, cfg_peleenetfrom utils.nms...

2020-02-25 23:16:10 1322

原创 Unsupported ONNX opset version: 11

Unsupported ONNX opset version: 11估计是pytorch不支持,重新安装pytorch2.0可能能解决。

2020-02-25 18:58:48 14484 2

原创 超越百度的口罩检测算法

思百达物联网科技(北京)有限公司算法团队开源了一款轻量级cpu版实时口罩检测项目。开源地址:https://github.com/sbdcv/sbd_mask人脸检测网络是centerface,是一款基于anchor free的高效的轻量级人脸检测网络。centerface widerface测试集的精度:easy set medium set hard s...

2020-02-25 10:37:24 7065

原创 No module named 'mxnet.gluon'

.No module named 'mxnet.gluon'原因:mxnet卸载就报错pip install mxnet==1.6.0

2020-02-24 23:46:25 3771

原创 No module named 'mxnet.gluon'

可知:cu100这个版本没有:mxnet-cu100

2020-02-24 18:15:26 3054

原创 cannot import name 'ImageRecordInt8Iter'

cannot import name 'ImageRecordInt8Iter'mxnet报错代码:from mxnet.gluon.model_zoo import visionfrom mxnet.gluon.nn import AvgPool2D

2020-02-24 18:05:23 2831

原创 deep_sort_face

deep_sort_facemosse:src/tracker/kalman_filter_track.py

2020-02-24 18:03:02 2318

原创 github添加图片

先把图片上传然后浏览器复制路径把tree改为raw然后按照下面的格式添加图片:在readme.md文件里添加对应的图片地址即可,保存即可显示![image]后面一窜就是图片地址![image](https://github.com/ButBueatiful/dotvim/raw/master/screenshots/vim-screenshot.jpg)...

2020-02-24 17:03:29 2698

原创 终止代码 page_fault_in_nonpaged_area

终止代码 page_fault_in_nonpaged_area电脑如果出现了“Page Fault In Nonpaged Area”错误,说明电脑的虚拟内存分页文件出现了问题,解决方法如下:第一步:用鼠标右键单击桌面上的“计算机”图标,在弹出的快捷菜单中选择“属性”。第二步:单击系统属性窗口里的“高级”标签。第三步:再单击性能旁边的“设置”按钮。第四步:出来的性...

2020-02-24 11:43:39 134585 14

原创 pvanet

https://github.com/sanghoon/pytorch_imagenet/blob/e76c2359b3361b6db0f9faf583487b2b1a784808/models/pvanet.pyimport timeimport torchimport torchvisionimport torch.nn as nnimport torch.nn.func...

2020-02-23 22:58:11 2534

原创 目标跟踪2020

https://github.com/jacke121/SiamMask效果不错,但是更像目标检测,办公室人体不能区分谁是谁。

2020-02-23 22:24:44 4097

原创 has no attribute 'xw_plus_b'

has no attribute 'xw_plus_b'解决:换成tf.compat.v1.nn.xw_plus_b

2020-02-23 11:37:32 2867

原创 Object arrays cannot be loaded when allow_pickle=False

Object arrays cannot be loaded when allow_pickle=False找到所在imdb.py中的所在行 将np.load(path)改为np.load(path, allow_pickle=True) 保存。重启Kernel

2020-02-23 11:31:40 2552

原创 reduce_sum() got an unexpected keyword argument 'keep_dims'

reduce_sum() got an unexpected keyword argument 'keep_dims'2.0.0版本出的问题:解决:normalize = tf.compat.v1.reduce_sum(target_exp, axis, keep_dims=True)softmax = tf.compat.v1.div(target_exp, normal...

2020-02-23 11:29:39 6463 5

原创 module 'tensorflow' has no attribute 'Session'

AttributeError: module 'tensorflow' has no attribute 'Session'错误解决原创陆羽飞 最后发布于2019-10-07 13:32:07 阅读数 16728 收藏展开Ubuntu刚安装好Tensorflow以后运行import tensorflow as tftf.Session()报错AttributeError: modu...

2020-02-23 11:17:52 9740

原创 深度学习工具软件

cuda 带显卡驱动cudnngitpython-opencv下载地址:http://www.lfd.uci.edu/~gohlke/pythonlibs/pycharmpytorchpaddlepaddle-gpunumpy

2020-02-23 10:51:24 2515

原创 face track

FaceTrackhttps://github.com/kylemcdonald/FaceTracker

2020-02-23 10:34:10 2618

原创 RuntimeError: cudnn64_7.dll not found.

paddlepaddle运行报错RuntimeError: cudnn64_7.dll not found.解决方法:把torch的dll拷贝到cuda的bin目录下。

2020-02-22 22:49:00 8817 1

原创 paddlepaddle安装

paddlepaddle安装

2020-02-22 22:38:36 3635 2

原创 PaddlePaddle yolov3

PaddlePaddle yolov3项目地址:https://github.com/PaddlePaddle/models/blob/v1.4/PaddleCV/yolov3/README_cn.md模型评估模型评估是指对训练完毕的模型评估各类性能指标。本示例采用COCO官方评估, 用户可通过如下方式下载Paddle发布的YOLOv3模型...

2020-02-22 19:03:11 3879

原创 No module named 'MySQLdb'

No module named 'MySQLdb'问题描述:项目在转到python3.7时,原先的导入MySQLdb模块都提示无法导入,pip install mysqldb也安装失败。问题原因:python2和python3在数据库模块支持这里存在区别,python2是mysqldb,而到了python3就变成mysqlclient,pip install mysqlcli...

2020-02-22 16:29:51 2441

原创 Facematch

Facematch虹软的:https://github.com/tyhjh/ArcFace

2020-02-22 12:15:19 2688

原创 frn

https://github.com/gupta-abhay/pytorch-frn原文:https://mp.weixin.qq.com/s?__biz=MzUyMjE2MTE0Mw==&mid=2247488803&idx=1&sn=c63149a91fefe07173a12271231f9a4b&chksm=f9d14bbbcea6c2adce882...

2020-02-21 22:16:26 2594

原创 网络优化方法

1.1*1 卷积层,谁需要谁用,不用在网络结尾用,2.dense_layer 252->8 有点变态,特征提取过程中太暴力3.通道数有必要增加吗?4.网络后面的参数别大于网络前面的参数,过滤比生成简单。...

2020-02-21 19:49:04 1244

原创 python 图像频谱

# -*- coding: utf-8 -*-import cv2 as cvimport numpy as npfrom matplotlib import pyplot as plt#读取图像img = cv.imread('test.png', 0)#快速傅里叶变换算法得到频率分布f = np.fft.fft2(img)#默认结果中心点位置是在左上角,#调用f...

2020-02-21 16:16:03 8032

原创 retinaface极坐标

import torchimport numpy as npdef point_form(boxes): """ Convert prior_boxes to (xmin, ymin, xmax, ymax) representation for comparison to point form ground truth data. Args: ...

2020-02-20 23:31:03 2434

原创 pytorch 忽略层权重

from collections import OrderedDictimport torchignor_names=["ssh1.conv7x7_3.1", "ssh2.conv3X3.1"]if __name__ == '__main__': trained_model=r"../weights/0.135.pth" state_...

2020-02-20 21:12:41 1324

原创 onnx 简化 onnxruntime

opencv dnn onnx 性能优化有时能达到16-10倍,onnx优化本身能达到10-14倍

2020-02-20 17:07:09 17763 5

原创 onnx onnxruntime 预测

速度比opencv 慢2ms左右,cpu占用从80降到50import osimport timeimport onnxruntimeimport numpy as npimport cv2import argparse#预处理图像,返回原尺寸图像,原尺寸,处理后图像数组def process_image(ori_img,input_shape): img =...

2020-02-20 16:31:16 1635

原创 tensor_proto.raw_data().empty() tensor_proto.float_data().empty() tensor_proto.double_data()解决

tensor_proto.raw_data().empty() || !tensor_proto.float_data().empty() || !tensor_proto.double_data().empty() || !tensor_proto.int64_data().empty()报错代码:# -*- coding: utf-8 -*-import onnximpor...

2020-02-20 14:04:23 3448

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

yolov5 openvino2022版本

yolov5 openvino 2022版本,win10系统vs2019,摄像头实时预览版本,亲测调试成功,特此给广大学友们分享

2022-05-22

yolov5 openvino版本

yolov5 openvino 2021版本,win10系统vs2019,摄像头实时预览版本,亲测调试成功,特此给广大学友们分享

2022-05-21

yolov5-face c++ mnn版

yolov5-face c++ mnn版,亲测成功, 3399设备上 100ms 320*240

2022-04-03

mnn windows centerface 示例项目

mnn center windows示例代码,包含mnn库,和centerface代码。

2022-02-17

yolov5各个版本 tensorrt cfg文件

yolov5各个版本 tensorrt cfg文件,下载地址: https://github.com/enazoe/yolo-tensorrt/blob/master/yolov5_tutorial.md 担心作者哪天不再共享,特地下载下来,分享给大家。

2022-02-16

car_select.rar

coco2017筛选提取车辆脚本,并转为xml格式,并手工精选车辆样本数据, 去除严重遮挡,暴露面积超级小的等等其他极难样本。

2022-01-13

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

coco2017转xml格式代码,并且手工挑选,去掉超级难度样本,留下精选样本列表 比如只有一只手,一只脚的样本; 剔除远处的超级样本,人体像素小于20个像素的超级小人体。 本人手工标注,花了3个周的时间,原创不易,请珍惜。

2022-01-13

ellipse_detector.rar

windows c++ opencv 圆检测,椭圆检测,源码,亲测可行

2021-12-26

Yolov5_DeepSort_bytetrack

Yolov5 DeepSort跟踪 bytetrack跟踪实例代码, deepsort 跟踪示例代码:track.py ByteTrack超越了此前所有的跟踪方法。 参考资料:https://blog.csdn.net/zhouchen1998/article/details/120932206 bytetrack示例代码:track_byte_demo.py

2021-11-20

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

yolov5 代码版 芒果检测 关于代码版,不是配置文件版,修改网络非常方便,参见我的博客: https://blog.csdn.net/jacke121/article/details/118392154 芒果检测 带训练权重,可以直接用

2021-11-16

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

旋转目标检测标签转换,支持以下数据格式转换为yolov5 txt 模式: dota 数据格式转换, itc_vd数据格式转换 uavrod_xml 数据格式转换 ucas数据格式转换 训练代码: https://github.com/BossZard/rotation-yolov5

2021-11-13

ByteTrack.rar

最强行人检测跟踪ByteTrack,原版代码有bug,输入视频,会出现把视频清空的bug,此为修正版,效果比yolov5强很多

2021-11-13

yolo系列 keams聚类anchors

yolo系列 keams聚类anchors

2021-10-29

yolov5_trackid.rar

yolo5 目标检测,行人、车辆等物体检测、跟踪、计数,车辆方向检测

2021-10-10

open3d_release_demo.rar

open3d 0.13最新版 win10 x64 release动态库,以及调用例子,亲测可用。

2021-09-18

mnn_yolofastest.rar

yolofastest mnn版,mnn1.2.0,win10系统,vs2019,亲测可用。

2021-09-18

yolov5-face_align.rar

yolov5-face 对齐人脸,yolov5-face 人脸检测,检测5个关键点,并对齐转正。 入库:detect_align.py

2021-09-01

mnn windows 转换工具 1.2.0

mnn windows 转换工具 1.2.0,release版,目前最新版,编译了2个小时,亲测可用,现在来分享给大家,可以直接用

2021-08-25

mnn1.2.0.zip

windows mnn x64位库 dll,lib文件,已经编译好了,直接用就行。

2021-08-24

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

2021-08-21

yolov5代码版,带预训练,测试脚本,可训练

yolov5配置版,阅读修改起来不太方便,特地整理出来代码封装版,阅读,修改网络比较方便,带预训练权重,检测代码:detect_class_s.py,检测芒果的,可以训练别的。

2021-08-21

labelimg_2021.rar

labelimg voc标注工具,自动生成图片文件夹,标注文件夹

2021-05-14

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

TA关注的人

提示
确定要删除当前文章?
取消 删除