pytorch
CV-deeplearning
小米AI Lab图像算法工程师,擅长表格识别、目标检测、行人属性识别、车牌识别等。
展开
-
Libtorch模型部署指南
前言:libtorch是pytorch原生支持的生态,理论上只要是pytorch训练的模型都能用libtorch部署,因为他们共用相同的c++代码。主要参考:Pytorch官网一. PyTorch模型转为Troch Scriptimport torchimport torchvision# An instance of your model.model = torchvision.models.resnet18()script_module = torch.jit.script(model原创 2022-05-18 17:25:08 · 1709 阅读 · 0 评论 -
图像去燥——TV Loss
1. 背景介绍 在图像复原过程中,图像上的一点点噪声可能就会对复原的结果产生非常大的影响,因为很多复原算法都会放大噪声。这时候我们就需要在最优化问题的模型中添加一些正则项来保持图像的光滑性,TV loss是常用的一种正则项(注意是正则项,配合其他loss一起使用,约束噪声)。图片中相邻像素值的差异可以通过降低TV loss来一定程度上解决。比如降噪,对抗checkerboard等等。2. 公式表达 即:求每一个像素和横向下一个像素的差的平方,加上纵向下一个像素...原创 2021-05-24 11:08:41 · 2342 阅读 · 0 评论 -
onnx 模型转换及推理时间对比
1. 环境准备 对比时间,和模型训练的环境相同,可能额外要安装的包是onnxruntime.pip install onnxruntime # for cpupip install onnxruntime-gpu # for gpu2. 测试过程 直接上代码吧,代码就是最好的解释。import cv2import timeimport torchimport numpy as npfrom torch.nn import DataParalle...原创 2021-04-29 20:53:44 · 4249 阅读 · 1 评论 -
基于fastreid训练行人属性模型
本文主要介绍用行人开源代码fastreid训练行人属性,行人属性数据集采用开源的PA00K。原创 2021-02-02 11:50:35 · 2836 阅读 · 10 评论 -
RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2
RuntimeError: Sizes of tensors must match except in dimension 1. Got 61 and 60 in dimension 2原创 2021-01-15 16:18:16 · 26651 阅读 · 3 评论 -
DOCKER里面PYTORCH关于GLOO地址声明 && NCCL地址声明
在docker中用多卡训练pytorch相关的模型时,GPU之间通常要进行通信,本文主要解决在docker中多卡训练报错问题。原创 2021-01-08 14:02:56 · 1458 阅读 · 0 评论 -
模型训练技巧——label smoothing
使用应用label smoothing在绝大多数分类任务上能提高模型的准确率,本文主要用pytorch实现label smoothing,使它能方便的应用到项目中。原创 2020-05-06 17:01:21 · 2357 阅读 · 2 评论 -
ResNeSt——ResNet最强改进版
张航、李沐等人提出ResNet最强改进版:性能提高3%,参数不增。是ResNet的最强改进版本,博主已经讲该网络应用与工程项目中。原创 2020-04-20 23:42:04 · 20354 阅读 · 39 评论 -
三分钟带你配置深度学习环境
conda可以理解为一个工具,也是一个可执行命令,其核心功能是包管理与环境管理。包管理与pip的使用类似,环境管理则允许用户方便地安装不同版本的python并可以快速切换。 三分钟带你用conda配置深度学习环境。原创 2020-04-13 18:57:19 · 594 阅读 · 0 评论