小恩子-CSDN博客

原创 API 接口压力测试

1、下载siege的安装包，我用的是siege-4.0.4.tar.gz 版本的。下载地址：http://download.joedog.org/siege/cd siege-4.0.4./configuremakesudo makeinstall2、测试 post接口siege -c 100 -t 10s -b 'http://192.168.***.***:7027/predict POST <./postfile.json'...

2022-03-02 17:24:50 1392

原创图像opencv base64 二进制互转

1、base64 opencv互转def image_to_base64(image_np): """ 将np图片(imread后的图片）转码为base64格式 image_np: cv2图像，numpy.ndarray Returns: base64编码后数据 """ image = cv2.imencode('.png', image_np)[1] image_code = str(base64.b64encode(image))[2:-1]

2022-02-11 09:39:29 1705

原创 c++实现nms算法

1、摘除每一类矩形框2、通过得分阈值排序3、通过计算iou排除大于iou阈值的矩形框，依次遍历。代码：#include<iostream>#include<vector>#include<algorithm>using namespace std;typedef struct Bbox { int x; int y; int w; int h; float score;}Bbox;static bool sort_score(B

2022-02-10 22:44:45 3247

原创 Tensorrt c++ api构建网络

路线：pytorch -> .wts->.engine重点：构建网络结构。参考：https://github.com/wang-xinyu/pytorchxhttps://github.com/wang-xinyu/tensorrtx代码：以alexnet为例。// Creat the engine using only the API and not any parser.ICudaEngine* createEngine(unsigned int maxBatch...

2022-02-08 17:55:01 2706

原创 c++遍历图像三种方式以及耗时比较

指针方式 at方式迭代器方式#include <iostream>#include <opencv2/core.hpp>#include <opencv2/highgui.hpp>using namespace std;using namespace cv;void ptr_cost(Mat img) { for (int i = 0; i < img.rows; ++i) { Vec3b* p =img.ptr<Vec3b&

2022-01-19 16:14:52 929

原创网络中加入注意力机制SE模块

SENet是由自动驾驶公司Momenta在2017年公布的一种全新的图像识别结构，它通过对特征通道间的相关性进行建模，把重要的特征进行强化来提升准确率。SENet 是2017 ILSVR竞赛的冠军。论文：Squeeze-and-Excitation NetworksSE block的基本结构给定一个输入，其特征通道数为C ，通过一系列卷积等一般变换后得到一个特征通道数为C的特征。 Squeeze:顺着空间维度进行特征压缩，将每个二维的特征通道变成一个实数，这个实数某种程...

2022-01-19 14:15:03 6832

原创 Pytorch Random Erasing

RandomErasing 随机擦除训练模型时，随机选取一个图片的矩形区域，将这个矩形区域的像素值用随机值或者平均像素值代替，产生局部遮挡的效果。该数据增强可以与随机切除、随机翻转等数据增强结合起来使用。在ReID、图像分类领域可以作为升点trick。codeimport cv2import mathimport randomimport torchvision.transforms as transformsfrom torch.utils.tensorboard import ..

2022-01-12 12:30:57 3043

原创 Pytorch tensorboard 可视化训练结果

训练模型时，导入 tensorboard. SummaryWriter 保存 loss、accuracy 等日志信息。code# 导入SummaryWriterfrom torch.utils.tensorboard import SummaryWriter# 实例化 SummaryWriterwriter = SummaryWriter(log_dir="./logs") # log_dir 日志存放路径# 模型训练时，写入train_loss、test_loss、train_ac

2022-01-09 18:13:28 756

原创网络过拟合处理方法

Missing sample data Early Stop Regularization Dropout Network Sructure1、Missing sample data 缺少样本数据过拟合问题万能解决办法就是增加样本数据，1、数据增广，2、增加采样。2、Early Stop 提前终止“早停止”很好理解，就是在validation的error开始上升之前，就把网络的训练停止了。这就是寻找最好模型的trick之一吧。如图虚线处停止训练3、Regulari...

2022-01-09 16:46:48 2408

原创 Warmup 模型训练之标配

学习率的调节策略等间隔调整学习率 StepLR 按需调整学习率 MultiStepLR 指数衰减调整学习率 ExponentialLR 弦退火调整学习率 CosineAnnealingLR 自适应调整学习率 ReduceLROnPlateau 自定义调整学习率 LambdaLRWarmup 使用然而在顶会论文和知名比赛中，作者一般都不会直接使用上述学习率调整策略，而是先预热模型（warm up）, warm up是深度学习炼丹时常用的一种手段，由于一开始参数不稳定，梯度较大，如

2022-01-07 19:41:14 1556

原创 c++ 图像尺寸不失真调整

原始图片尺寸：744*496输入416*416 网络，图片尺寸失真调整：输入416*416 网络，图片尺寸不失真调整：c++ 代码:#include <iostream> #include <vector>#include <opencv2/opencv.hpp>using namespace cv;static const int INPUT_H = 416;static const int INPUT_W = 416;..

2021-12-27 17:49:33 2819

原创 Pytorch模型转为onnx,onnx 模型 inference测试

1、mobilenet_v2生成.pt用torchvision导出mobilenet_v2网络结构，生成mobilenet_v2.ptimport torchfrom torch import nnfrom torchvision import modelsimport torch.nn.functional as Fclass MobileNet_v2(nn.Module): def __init__(self): super(MobileNet_v2, sel

2021-12-25 20:08:16 1425

原创车辆属性识别

车辆属性车辆检测识别图像中的所有车辆，返回每辆车的类型和坐标位置，识别小汽车、卡车、巴士、摩托车、三轮车、自行车等多类车辆。车辆属性识别针对汽车识别多种外观属性，包括：是否有车窗雨眉、是否有车顶架、副驾驶是否有人、驾驶位是否系安全带、遮阳板是否放下、车辆朝向等，针对不同数据集可以做不同种类属性分析。本文采用VeRi的车辆重识别的大规模基准数据集。VeRi的特点包括：20台摄像机在1KM^2的城市区域内24小时拍摄而来，包含776辆车的超过50,00...

2021-12-24 11:17:10 3134 4

原创人体属性识别

人体属性检测图片中所有的人体，可针对每个检测出的人体，进行人体属性分析，识别人体的性别年龄、衣着外观等特征，辅助定位追踪特定人员。人体检测检测图片中所有的人体，检测人体的矩形框位置。人体属性采用PA-100K dataset数据集训练，包含10000个行人100000张图像，26种属性，本文使用如下10种属性。gender：'男', '女'age:"老年", "中年", "少年"orientation:'正面', '侧面', '背...

2021-12-24 10:43:27 2560 2

原创人脸属性识别之二

人脸属性人脸是一种非常重要的生物特征，具有结构复杂、细节变化多等特点，同时也蕴含了大量的信息，比如性别、种族、年龄、表情等。一个正常的成年人可以轻易的理解人脸的信息，但将同样的能力赋予给计算机，并让其代替人类进行类脑思考成为研究学者亟待攻克的科学课题！人类可以通过使用相机等图像采集装置和计算机组建一套与人体类似的系统，相机等图像采集装置是“眼睛”，计算机是“大脑”。但是问题来了，这些单纯的硬件设施并不足以让机器完成理解人脸信息的任务，这其中还需要载有思考能力，也就是我们平...

2021-12-24 10:00:02 1599 1

原创 Python3.7之后，sklearn 的linear_assignment不支持，用scipy代替

import numpy as npfrom sklearn.utils.linear_assignment_ import linear_assignmentfrom scipy.optimize import linear_sum_assignment as linear_sum_assignmentconst_matrix=np.array([[15,40,45],[20,60,35],[20,40,25]])matches=linear_assignment(const_matrix).

2021-11-30 18:16:51 2330 4

原创 1.fatal error: NvInfer.h: No such file or directory

编译yolov5生成engine，报/tensorrtx/yolov5/yololayer.h:6:10: fatal error: NvInfer.h: No such file or directory6 | #include <NvInfer.h> | ^~~~~~~~~~~compilation terminated.1、未找到tensorRT对应的头文件和库。在CMakeLists.txt#tensorrt处添加tensorrt对应头文件和...

2021-11-22 09:25:30 3717

原创 ubuntu升级python和pip

由于各种原因需要升级python到更高版本，同时需要建立pip和python对应映射。推荐两个博客：1、在Ubuntu 16.04(18.04) 系统上编译安装Python 3.8 方法2、ubuntu16.04纯净版-安装Python3.8.1/升级pip

2021-11-21 22:19:16 1023

原创计算pytorch模型算力和参数的大小

设计神经网络时需要根据硬件考虑模型参数量和算力，故需要计算神经网络模型的params和Flops，python的三方库ptflops就可以很好的做这件事，show code：import torch.nn as nnimport torchfrom ptflops import get_model_complexity_infoclass Net(nn.Module): def __init__(self,in_c,class_num): super(Net, self)

2021-10-17 21:22:43 1829

原创人脸属性识别之一

人脸属性识别，包括年龄、性别、种族等属性

2021-08-22 22:30:19 712 4

原创 TX2 pytorch 安装

TX2 jetpack4.5 对应cuda10.2,需要与pytorch对应，不然没法使用，版本对应如下。TX2 torch-1.7.0.0-cp36-cp36m1-linux_aarch64.whl 百度网盘下载：链接：https://pan.baidu.com/s/1lI1opYAdryJQBrNsZLn63w 提取码：ovsg ...

2021-06-23 10:49:57 294

原创 Pulsar 订阅存在bug

1、最近在用pulsar开发项目，在项目中不断关闭、开启线程，每一次开启线程都要订阅一个摄像头的topic。发现出现内存泄漏。import pulsar,timeclient = pulsar.Client("pulsar://15.18.162.112:6650")consumer =client.subscribe("non-persistent://public/default/topic-" + "15010200001310004341**", 'smoke-subscription')

2021-06-22 16:54:42 418

原创 JETSON TX2 刷机注意事项

最近手上有一块JETSON TX2,网上有很多详细教程就不具体了，但有的教程有误导性。1、host machine可以不勾选，交叉编译使用的，我就没有勾选，jiandi

2021-06-16 14:49:08 141

原创 please determine the input size manually by “--dynamic-input-shape --input-shape xxx“ or “--input-sh

最近在做onnx转trt使用onnxsim出现这个错误，源代码如下：from onnxsim import simplifyimport onnxinput_path="cnn.onnx"output_path="sim_cnn.onnx"onnx_model = onnx.load(input_path) # load onnx model)model_simp, check = simplify(onnx_model)assert check, "Simplified ONNX mo

2021-06-05 11:44:37 1815

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

1、pytorch模型转为onnx后用tensorRT量化加速时，报这个问题。OnnxParser在解析onnx文件时没解析成功,这个问题一方面可能涉及到pytorch和tensorrt版本的问题,另一方面可能涉及到网络模型有些操作, tensorrt中的onnx解析器不支持,解析不了.第一方面, tensorrt6.0系列只能使用pytorch1.2及以下版本. 所以需要使用torch1.2生成onnx文件.第二方面, 就要将一些不支持的操作进行替换, 比如expand_as等操作有时候上

2021-06-04 11:21:12 1297 2

原创 keras 实现vgg6 网络训练fashion_mnist数据

keras应该是所有框架中最简洁友好的。import tensorflow as tfimport osimport numpy as npfrom matplotlib import pyplot as pltfrom tensorflow.keras.layers import Conv2D, BatchNormalization, Activation, MaxPool2D, Dropout, Flatten, Densefrom tensorflow.keras import Mod

2021-05-18 22:07:55 386

原创 device CUDA:2 not supported by XLA service while setting up XLA_GPU_JIT device number 2

torch运行的时候,每一块显卡都会占用一点显存,如果只想让其运行在指定的显卡上,可以使用下面的命令:export CUDA_VISIBLE_DEVICES=1,2在执行torch程序的时候,先在终端中执行上面的命令,就可以让该终端只看到指定的显卡,上面的命令中的1和2就是nvidia-smi命令看到的显卡的序号.执行了上面的命令后,torch指定的GPU从0开始按照上面可见的依次往后....

2021-04-30 11:32:23 2308

原创 RuntimeError: unable to write to file ＜/torch_2818186_3541370554＞ ERROR: Unexpected bus error encoun

在容器中，训练yolov5时，报如上问题。pytorch将共享内存的临时文件保存在了/torch_xxx文件中，即容器中的根目录下。容器磁盘空间不足导致该问题的发生。目前可以通过以下代码暂时关闭pytorch的shared memory功能来规避：将此代码贴到你启动代码的最前面，顺利解决：import sysimport torchfrom torch.utils.data import dataloaderfrom torch.multiprocessing import reductio

2021-04-26 11:23:36 1330

原创 AttributeError: ‘list‘ object has no attribute ‘dtype‘

AttributeError: 'list' object has no attribute 'dtype' tensor_info_output = tf.saved_model.utils.build_tensor_info(model.output)改为 tensor_info_output = tf.saved_model.utils.build_tensor_info(model.output[0])

2020-11-11 18:22:11 3762

原创 TypeError: init() got an unexpected keyword argument ‘ragged‘

TypeError: __init__() got an unexpected keyword argument 'ragged'from keras.models import load_model 改为from tensorflow.keras.models import load_model

2020-11-11 18:00:30 1341

原创 AI 展示（3）打电话识别

1、爱情公寓片段打电话识别

2020-07-15 23:22:46 2778 1

原创 AI 展示（2）目标跟踪

1、车目标跟踪车目标跟踪2、车目标跟踪人等目标跟踪

2020-07-15 23:20:00 155

原创 AI展示（1）目标检测

1、舞蹈片段目标检测2、电影功夫片段目标检测（功夫片段）

2020-07-15 23:04:08 145

原创使用imutils包的VideoStream读取视频流，丢帧率达20%

最近在读取相机RTSP流，发现OpenCV不如下面方式读取的快：import _threadfrom imutils.video import VideoStreamimport imutilsimport timeimport numpy as npimport cv2# # 为线程定义一个函数def cam1( threadName): cap = VideoStream('test.mp4').start() n=0 while True:

2020-07-14 15:42:13 5106 3

原创 Ubuntu配置添加环境变量

在python/C++/C编译时需要加载动态库，需要添加环境指定库的位置：其实有三种，掌握其中一种即可：1、vim ~/.bashrc2、在.bashrc尾部添加一行export PATH=$PATH:/home/******/******3、source ~/.bashrc 生效...

2020-06-30 21:30:53 734

原创 fatal error: Python.h: 没有那个文件或目录 compilation terminated.

pythonn 3.5 遇上如上问题，网友给的解决方案：sudo apt-get install python3-dev未解决：sudo apt-get install python-dev解决！！！

2020-06-30 21:13:12 1636

原创 python gRPC (1)

RPC基础RPC 是两个子系统之间进行的直接消息交互，它使用操作系统提供的套接字来作为消息的载体，以特定的消息格式来定义消息内容和边界。gRPC 是 Google 开源的基于 Protobuf 和 Http2.0 协议的通信框架，Google 深度学习框架 tensorflow 底层的 RPC 通信就完全依赖于 gRPC 库gRPC的使用通常包括如下几个步骤：通过protobuf来定义接口和数据类型编写gRPC server端代码编写gRPC client端代码下面来通过一个

2020-06-27 21:02:13 193

原创 ModuleNotFoundError: No module named numpy.random._pickle

原因：numpy版本较老，版本不匹配。解决：pip install numpy --upgrade --user

2020-06-10 11:26:58 6918 2

原创 TypeError: namedtuple() got an unexpected keyword argument -verbose

File "C:\Users\Desktop\Online-Realtime-Action-Recognition-based-on-OpenPose-master\Pose\pose_estimator.py", line 21, in <module> 'coord1', 'coord2', 'score1', 'score2'], verbose=False)TypeError: namedtuple() got an unexpected keyword argument 'v..

2020-06-03 09:06:41 3127

原创 graph_def.ParseFromString(f.read()) google.protobuf.message.DecodeError: Error parsing message

当测试https://github.com/ildoonet/tf-pose-estimation/ 仓库执行如下命令时python run.py --model=cmu --resize=656x368 --image=./images/p1.jpg遇到如上问题，在网上没有找到解决方案，仔细分析原因：在下载模型时发现下载的模型不全只有几百KB大小;通过如下命令很难把模型完整下载下来。$ cd models/graph/cmu$ bash download.sh模型下载链接：链接: .

2020-05-30 19:47:52 5526