AI
AI大魔王
这个作者很懒,什么都没留下…
展开
-
blobFromImage
blobFromImage原创 2022-03-15 10:40:54 · 1340 阅读 · 0 评论 -
YOLOX实战使用技巧
mosaic该操作是将4张图片进行拼接成一张大图;所以设置为了矩形640*640;如果去除该操作那么生成的图片就是原创 2022-02-24 14:19:55 · 1751 阅读 · 1 评论 -
YOLOX训练自己的数据集
github代码论文准备环境conda create -n YOLOX python=3.8conda activate YOLOXpip install torch==1.8cd YOLOXpip install -r requirements.txt2、安装pycocotools下载地址cd pycocotools-2.0.2python setup.py build_ext install3、安装apex这个工具为什么要安装呢,提高你的显存能力,比如你的显存.原创 2021-09-02 15:18:28 · 1058 阅读 · 1 评论 -
pytorch 最全optimizer
SGDcodeimport torchoptimizer = torch.optim.SGD(params, lr=0.001, momentum=0, dampening=0, weight_decay=0, nesterov=False)Adamcodeimport torchoptimizer = torch.optim.Adam(params, lr=0.001, betas=(0.9, 0.999), eps=1e-08, weight_decay=0, amsgrad=False原创 2021-08-18 10:50:13 · 949 阅读 · 0 评论 -
pytorch scheduler汇总
ExpLR指数衰减学习率按照指数的形式衰减是比较常用的策略,我们首先需要确定需要针对哪个优化器执行学习率动态调整策略,其中参数gamma表示衰减的底数,选择不同的gamma值可以获得幅度不同的衰减曲线codeoptimizer_ExpLR = torch.optim.SGD(net.parameters(), lr=0.1)ExpLR = torch.optim.lr_scheduler.ExponentialLR(optimizer_ExpLR, gamma=0.98)StepLR原创 2021-08-17 10:08:04 · 8051 阅读 · 2 评论 -
DynamicReLU
论文该文是MSRA的研究员在DynamicConv之后,又一次在激活函数方面的改进。相比静态激活函数,动态激活函数可以将MobileNetV2的性能由72.0%提升到76.2%代价仅是增加额外的5%FLOPS。从这个角度来看,这个代价还是非常值得的。辅助函数:它用于计算激活函数的参数; 激活函数:它用于计算输入的激活输出,它的参数通过上述辅助函数生成。需要注意的是:辅助函数对输入进行上下文信息编码并用于辅助后续激活函数。这种数据依赖性使得所提DyReLU可以极大的提升模型的表达能能力,尤原创 2021-08-16 13:05:08 · 321 阅读 · 0 评论 -
PyTorch使用TensorboardX进行训练可视化
demo代码if __name__ == '__main__': out="/home/zengxh/workspace/zdata" summarywriter=run_tensorboard(out,port=6006) # port为0的话是生成不冲突的随机端口 # 记录模型 model = torchvision.models.shufflenet_v2_x0_5() dummy_input = torch.randn(1, 3, 224, 224)原创 2021-08-14 10:05:57 · 209 阅读 · 0 评论 -
SWA(Stochastic Weight Averaging)实验
原理论文地址:https://arxiv.org/pdf/2012.12645.pdfSGD倾向于收敛到loss的平稳区域 平稳区域的大部分都处于边界,由于权重空间的维度比较高,SGD通常只会走到这些平稳区域的边界;SWA通过平均多个SGD的权重参数,使其能够达到平稳区域的中心.有论文说swa能涨分,那么我来实验一下那么我将在cifar10数据集上进行实验大致的原理就是:找最后几个epoch的平均实验模型采用:shufflenet_v2_x0_5数据集.原创 2021-08-13 14:03:49 · 7202 阅读 · 1 评论 -
PyTorch: 训练分类CIFAR10
# !/usr/bin/env python# -- coding: utf-8 --# @Author zengxiaohui# Datatime:8/13/2021 11:20 AM# @File:train_cifar10import osimport torchimport torchvisionimport torchvision.transforms as transformsimport torch.optim as optimimport torch.nn as nn.原创 2021-08-13 13:12:35 · 215 阅读 · 0 评论 -
Swin-Transformer-Object-Detection训练自己的数据集
环境配置conda create -n SwinTransformerObjectDetection -y python=3.8conda activate SwinTransformerObjectDetectionconda install pytorch=1.8.0 torchvision cudatoolkit=10.2 -c pytorch -y# mmcv安装cd ~/software/rm -rf mmcvgit clone https://github.com/open-原创 2021-05-27 17:05:31 · 5750 阅读 · 14 评论 -
CenterNet2训练自己的数据集
centernet2训练自己的数据集环境配置1.下载https://github.com/xingyizhou/CenterNet22.安装环境conda create -n CenterNet2 python=3.8conda activate CenterNet2pip install torch==1.8python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'pip insta原创 2021-05-12 16:18:01 · 10219 阅读 · 13 评论 -
opencv decode加速
速度慢到快start = time.time() img = cv2.imdecode(np.fromstring(img, dtype=np.uint8),-1) print(time.time() - start) start = time.time() aa = cv2.imdecode(np.frombuffer(file_read, dtype=np.int8), cv2.IMREAD_UNCHANGED)原创 2021-04-16 16:17:31 · 1403 阅读 · 0 评论 -
pytorch to_categorical
y = pred.flatten() categorical = torch.zeros((65536, 10), dtype=torch.int) categorical[torch.arange(65536), y] = 1 categorical = torch.reshape(categorical, (256, 256, 10))原创 2021-04-16 11:19:41 · 2232 阅读 · 0 评论 -
深度玄学-实战开发步骤
debug式的开发深度学习项目的步骤只要做好其中的每一步,那么做出来的项目一定是很适用的原创 2021-03-24 10:57:41 · 163 阅读 · 0 评论 -
argmax的区别
numpy 的argmax和torch1.7.0 的argmax 的结果一样但是torch1.6.0的argmax的结果和上面2个都不一样原创 2021-03-12 09:23:20 · 192 阅读 · 0 评论 -
pytorch 不能导出yolov5的onnx
if opname == 'hardswish': relu=_registry[(domain, version)]["relu"] _registry[(domain, version)][opname] = relu原创 2021-02-18 09:57:28 · 551 阅读 · 0 评论 -
模型训练优化总体步骤
数据预处理分析数据训练中间过程分析(将中间层的结果打印成图片,分析)模型预测结果后处理分析原创 2021-02-18 09:37:10 · 591 阅读 · 0 评论 -
python SMBClient 读取图片
import ioimport cv2from PIL import Imagefrom smb.SMBConnection import *import numpy as npclass SMBClient(object): """ smb连接客户端 """ prot = None status = False samba = None def __init__(self, user_name, passwd, ip, port=1...原创 2020-09-02 15:20:16 · 1518 阅读 · 0 评论 -
udp,ftp,smb,socket传输速度对比
udp: 传输速度最快,但是不保证传输质量ftp: 比smb,http要快,但是对于1M以下的图片 http要快smb: 大一些的图片比http要快socket:是一种连接协议,不负责传数据,需要用socket搭建成tcp才可以传ftp,sftp:都是用tcp实现的终极传输,史上最快:多线上传,压缩上传,压缩+多线上传上述传输的极限都是带宽大小,多线上传这个不知道...原创 2020-05-27 18:28:19 · 13426 阅读 · 1 评论 -
ubuntu关机开机后显卡挂了:报错NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. M...
电脑重启后nvidia-smi启动不了whereis nvidiayum --enablerepo=epel install dkmssudo dkms install -m nvidia -v 410.129原创 2020-05-09 17:57:45 · 315 阅读 · 0 评论