![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Coding
文章平均质量分 67
talk is cheap, show me the code
lokvke
It is a secret.
(不定时更新)
展开
-
一种非极大值抑制(non_max_suppression, nms)的代码实现方式
非极大值抑制的实现原创 2022-07-20 01:19:40 · 1182 阅读 · 1 评论 -
python保存h264格式视频(linux和windows)
python保存h264格式视频(linux和windows)原创 2022-06-28 09:16:32 · 5632 阅读 · 1 评论 -
Pytorch模型加密的方法
目录一、难点和原理二、代码三、整体代码四、硬件绑定一、难点和原理对一个文件进行加密或者解密其实有很多种方法,难点主要在于,得到加密文件之后,如何在程序运行时进行解密,而不是先解密再运行程序,这相当于没加密。(脱裤子放屁的感觉。。。)对于pytorch框架,我们进入torch.load里面查看代码,如下图所示:pytorch是可以读取 io.BytesIO 类型的数据的,因此,可以有如下的加密和解密过程:加密过程我们可以将读取的模型先保存成 io.BytesIO的格式;再把io.By原创 2022-04-24 21:05:10 · 11627 阅读 · 27 评论 -
手写深度学习——numpy实现卷积层
numpy实现卷积层原创 2022-04-04 21:55:10 · 3037 阅读 · 0 评论 -
python打印日志,同时保存在文件(解决log重复打印问题)
python的log日志,打印在终端同时保存在文件,解决重复打印问题原创 2022-04-02 10:05:44 · 1856 阅读 · 2 评论 -
opencv在图像上绘制直线或指定区域
opencv在图像上绘制直线或指定区域原创 2022-02-28 10:04:19 · 6452 阅读 · 0 评论 -
faiss的python接口使用
faiss的python接口使用1. 简介2. 安装3. 示例1. 简介faiss是一种ann(Approximate Nearest Neighbor)库,可以用于特征的入库,检索。不仅可以在cpu上使用,还可以利用GPU进行检索,提高检索的速度。具体可以参考:https://github.com/facebookresearch/faiss2. 安装cpu版本,适用于各个系统pip install faiss-cpucpu + gpu版本,目前不适用于windows系统pip ins原创 2021-08-10 11:22:54 · 9566 阅读 · 0 评论 -
YOLO数据格式说明与转换
YOLO数据格式说明与转换1. YOLO数据格式说明2. 代码1. YOLO数据格式说明yolo标注格式保存在.txt文件中,一共5个数据,用空格隔开,举例说明如下图所示:假设图像的高和宽分别为h, w,bbox的左上角坐标为(x1, y2),右下角坐标为(x2, y2),则可求得bbox中心坐标(x_c, y_c)为:x_c = x1 + (x2 - x1)/2 = (x1 + x2)/2y_c = y1 + (y2 - y1)/2 = (y1 + y2)/2假设yolo的5个数据分别为原创 2021-07-09 10:57:13 · 28373 阅读 · 1 评论 -
flask接收数据流(图像文件、视频文件)的几种方式
flask接收数据流(图像文件、视频文件)的几种方式1. 接收上传的图像文件2. 接收上传的视频文件1. 接收上传的图像文件flask的接口定义如下,接收单张图像,返回检测结果。@app.route('/detect', methods=['POST'])def detect_api(): try: file = request.files.get("file") if file is not None: detections = de原创 2021-06-10 10:29:14 · 9786 阅读 · 10 评论 -
Flask部署深度学习模型的一个例子
Flask部署深度学习模型的一个例子一. Flask二. postman工具一. Flask工程一共包括三个文件:用于测试的图片1.jpgindex_to_name.py定义了一个字典,键是预测类label,值是对应的class_name,如下图所示:主要代码文件app.pyfrom index_to_name import img_class_mapimport torchvision.models as modelsimport torchvision.transf原创 2021-05-19 18:45:49 · 1513 阅读 · 3 评论 -
opencv把一系列图像保存为视频、一系列图像保存为gif
目录一. 图像转为视频二. 读取摄像头一. 图像转为视频import cv2import ossave_name = 'demo.avi'fourcc = cv2.VideoWriter_fourcc(*'DIVX')fps = 30width = 1920height = 1080out = cv2.VideoWriter(save_name, fourcc, fps, (width, height))img_dir = r'D:\data\2DMOT2015\test\ADL-R原创 2021-04-08 15:57:51 · 2569 阅读 · 1 评论 -
pytorch实现分类遇到的一些问题
目录一. 利用torchsummary打印模型结构二. 模型定义结尾要不要接入softmax?三. 训练时计算loss为什么要乘以batch_size?一. 利用torchsummary打印模型结构安装torchsummarypip install torchsummary代码如下:import torch.nn as nnimport torch.nn.functional as Ffrom torchsummary import summaryclass Net(nn.Mo原创 2021-03-24 18:21:03 · 700 阅读 · 0 评论 -
faster-rcnn中RPN网络的anchor实现:numpy和pytorch版本
目录一.pytorch版本二. numpy实现三. 总结一.pytorch版本pytorch版本为1.7.1sizes和aspect_ratio设置如下代码所示:from torchvision.models.detection.rpn import AnchorGeneratoranchor_generator = AnchorGenerator(sizes=((32, 64, 128),), aspect_ratio原创 2021-03-08 22:08:22 · 439 阅读 · 2 评论 -
Python利用fitz库提取pdf中的图片(针对多种类型pdf)
目录一. 安装fitz二. pdf文件格式问题2.1 pdf文件存在多种格式2.2 分析问题三. 代码一. 安装fitz安装:需要安装fitz和PyMuPDF,否则会报如下错误:ModuleNotFoundError: No module named ‘frontend’pip install fitz PyMuPDF二. pdf文件格式问题2.1 pdf文件存在多种格式pdf文件的格式有好几种,用Adobe Acrobat比较正常的如下所示:这种类型的pdf文件可以比较正常地提取里面的图片原创 2020-09-23 09:41:30 · 24788 阅读 · 5 评论 -
IoU的计算
目录一. IoU介绍二. 计算思路2.1 相交的情况2.2 不相交情况三. 代码一. IoU介绍IoU,即 intersection over Union,就是两个矩形框的交集面积与他们并集面积的比值。IoU也是一种算法性能的指标,例如在语义分割时就会用IoU来衡量分割效果的好坏。举例说明,如下图所示:已知:左边的矩形框假设为N,左上角坐标为(x1, y1), 右下角坐标为(x2, y2);右边的矩形框假设为M,左上角坐标为(a1, a1), 右下角坐标为(a2, a2);相交的矩形框原创 2020-08-04 14:56:41 · 10644 阅读 · 3 评论 -
CrossEntropy的numpy实现和Pytorch调用
目录1. Numpy实现2. Pytorch实现3. nn.CrossEntropy的weight参数1. Numpy实现import torchimport numpy as npfrom torch.nn import functional as F# 定义softmax函数def softmax(x): return np.exp(x) / np.sum(np.exp(x))# 利用numpy计算def cross_entropy_np(x, y): x_soft原创 2020-07-31 16:08:17 · 5303 阅读 · 2 评论 -
python——distance_transform_edt的使用(计算距离)
目录1. distance_transform_edt介绍2. 代码3. 应用——计算图像中两个同心圆之间的区域1. distance_transform_edt介绍distance_transform_edt是scipy库里的一个函数,用于距离转换,计算图像中非零点到最近背景点(即0)的距离。举例说明:(1)现有图像A,如下图所示(2)离最近背景点距离为1的用红色标出,如下图所示:(3)离最近背景点距离为2\sqrt 22 的用绿色标出,如下图所示:(4)离最近背景点距离为2 的用橘色原创 2020-06-08 00:22:09 · 12228 阅读 · 9 评论 -
numpy实现zero-padding
import numpy as np"""input:[[1, 2, 3, 4, 5], [4, 5, 6, 7, 8],[9, 8, 7, 6, 5],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6],[2, 3, 4, 5, 6],[2, 3,...原创 2019-08-30 17:29:51 · 4250 阅读 · 6 评论