![](https://img-blog.csdnimg.cn/20201014180756918.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
一名CV界的小学生
这个作者很懒,什么都没留下…
展开
-
STDC分割网络:onnx推理
def STDCProcess(img): mean = [0.485, 0.456, 0.406] std = [0.229, 0.224, 0.225] img = cv2.resize(img, (256, 256)) img = img.astype(np.float32) img = img[:, :, ::-1]# BGR2RGB img = (img / 255.0 - mean) / std img = np.transpose(i.原创 2022-01-17 16:50:47 · 2826 阅读 · 0 评论 -
离线数据增强
def imgAug(imgDir,dstImg): params = [0.5, 0.6, 0.7, 0.8, 0.9, 1.1, 1.2, 1.3, 1.4, 1.5] gaussParams=[3,5] imgNames=[img for img in os.listdir(imgDir)] for name in imgNames: img = Image.open(os.path.join(imgDir,name)) brightE.原创 2021-12-28 17:37:22 · 1642 阅读 · 0 评论 -
pytorch API:随机裁剪
def imgAugTest(imgDir): img=Image.open(imgDir) print('the size of raw img:{}'.format(img.size)) cropImg=transforms.RandomResizedCrop(224)(img) print('the size of new img {}'.format(cropImg.size)) cropImg01=transforms.RandomResizedCrop(.原创 2021-12-17 15:46:04 · 1045 阅读 · 0 评论 -
Pytorch API:学习率调整
def StepLRTest(): model=AlexNet(num_classes=2) optimizer=optim.SGD(params=model.parameters(),lr=0.01) scheduler=lr_scheduler.MultiStepLR(optimizer,milestones=[5,20,40,80],gamma=0.1) plt.figure() x=list(range(100)) y=[] for ep.原创 2021-12-13 16:32:38 · 1412 阅读 · 0 评论 -
YOLO:labelImg标注的xml转成yolo格式
代码备忘:# -*- coding: utf-8 -*-import xml.etree.ElementTree as ETimport osfrom os import listdirfrom os.path import joinimport randomclasses = ["person"] # 自己数据集有哪些类别写哪些类,按照顺序mainDir=r'/home/.../extra/personDataset'def convert(size, box): dw原创 2021-10-26 17:20:31 · 815 阅读 · 0 评论 -
pytorch: 查看特征图
其中网络结构为:GitHub - chenhuanan/person_search_demo: 利用YOLOv3结合行人重识别模型,实现行人的检测识别,查找特定行人from reid.modeling import build_modelfrom reid.config import cfg as reidCfgimport torchimport cv2import numpy as npimport torchvision.transforms as transformsimpo.原创 2021-09-23 15:10:17 · 527 阅读 · 1 评论 -
conda / pip 命令 备忘
创建conda环境conda create -n your_env_name python=xxx其中:your_env_name 虚拟环境名称 python=xxx 指定python版本例如:conda create -n nlp python=3.6另:虚拟环境地址在anaconda的安装目录下的envs文件夹下...原创 2021-06-08 17:01:16 · 136 阅读 · 1 评论 -
实时监视显存
实时显示显存信息指令:nvidia-smi -lms=100 //每 100 ms 刷新一次或:watch -n 0.1 nvidia-smi //每 1000 ms 刷新一次参考:https://blog.csdn.net/C_chuxin/article/details/82993350 nvidia-smi 表格内容解析原创 2021-03-26 15:15:19 · 524 阅读 · 0 评论 -
深度学习 训练集与验证集损失分析
训练集损失下降,验证集损伤下降 ——>网络正在学习(理想状态)训练集损失下降,验证集损失不变——>网络过拟合(尝试drpout,L2正则化等手段)训练集损失不变,验证集损失下降——>数据集有问题(检测数据集)训练集损失不变,验证集损失不变——>网络学习遇到瓶颈(减小学习率learning_rate或batch_size)训练集损失上升,验证集损失上升——>网络结构设计不当、超参数设置不当、数据集经过清洗等参考:https://zhuanlan..原创 2021-03-13 23:10:55 · 8176 阅读 · 0 评论