![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
文章平均质量分 60
多为日常环境搭建、模型训练操作
李昂的
这个作者很懒,什么都没留下…
展开
-
yolov5中workers设置问题
在每个epoch加载数据时,dataloader一次性创建workers数量个工作进程,并且每个进行负责调度一个batch的数据,worker负责将batch加载到内存中,dataloader从RAM中找到这个epoch所需要的batch,如果找到就使用,如果没有就要继续加载batch到RAM。所以worker的设定需要根据自己的服务器设备情况和数据情况设定,如果CPU和RAM很强,就可以设置大一些,否则就需要设置小一些,如果单张图像很大,也需要设置小一些。原创 2023-03-05 22:15:25 · 3716 阅读 · 1 评论 -
解决使用keras训练模型Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR报错问题
根据上面分析的原因修改了batchsize也是异常,然后查找了一下相关资料,一般情况下,如果不加以限制,很多深度学习框架的代码在运行的时候会申请整个显存空间,即便它不需要这么多的资源,但是它申请了之后就不允许其他的程序使用,所以在这种状态下运行代码,就会出现显存不够用的问题,所以只要分配一下模型训练的显存分配策略就可以。使用keras训练模型时异常报错,根据日志分析大概是显存益处问题,通过watch -n 1 nvidia-smi 观察显存使用情况,也发现当模型训练时显存会突然爆增,然后进程就挂掉了。原创 2022-11-14 21:45:40 · 424 阅读 · 0 评论 -
解决pytorch中tensorboard:No dashboards are active for the current data set问题
后始终报错,开始各种查资料,包括语法、tensorboard版本问题,都无法解决,最后发现了一个愚蠢的问题,tensorboard中的 logdir路径是相对路径,在pycharm中terminal的是工程路径,而我的代码是在工程下的一级目录里面,固然找不到生成的日志文件,CD到运行的代码上一级目录解决问题。在pycharm中训练模型,使用terminal开启。在pytorch中tensorboard使用方法。原创 2022-11-07 00:19:07 · 572 阅读 · 0 评论 -
nvidia显卡驱动、cuda、cudnn、tensorflow对应版本
在nvidia官网下载驱动,驱动官网选择设备的驱动进行搜索下载即可,搜索时注意对应的操作系统一般为安装NVIDIA Studio驱动版本,GeForce Game Ready适用于游戏玩家,下面是两个版本区别的官方解释下载完的驱动会以版本号的形式命名,这个版本号在安装cuda时候会用到,以下是下载好的驱动文件名516.93-desktop-win10-win11-64bit-international-nsd-dch-whql.exe516.93–显卡驱动版本号desktop–原创 2022-10-26 22:49:25 · 7067 阅读 · 0 评论 -
图像处理小知识
载入图像滤波降噪它只取内核区域下所有像素的平均值并替换中心元素,使用n*n的滤波器对图像进行卷积操作,使图像变为模糊,卷积核越大,图像越模糊。转为灰度图进行开闭运算腐蚀运算:移动结构B,如果结构B与结构A的交集完全数据结构A的区域,则保留该点。膨胀运算:将机构B在结构A上进行卷积操作,移动结构B的过程中,与结构A存在重叠区域,则保留该点。开运算:先进行腐蚀运算,在进行膨胀运算,可以消除一些小东西(亮的),放大局部低亮度区域,平滑较大物体边缘。1、能够去处孤立的小点、毛刺,图像整体的形状不原创 2021-11-05 11:25:06 · 406 阅读 · 0 评论 -
mmdetection中使用自己的数据测试
mmdetection中进行测试使用自己的数据进行测试,保存到本地并且能够按照置信度进行区分,测试代码如下。#!/usr/bin/env python3# -*- coding: utf-8 -*-from argparse import ArgumentParserimport osfrom mmdet.apis import inference_detector, init_detector #, show_result_pyplotimport cv2import timeimpo原创 2021-08-04 16:44:53 · 901 阅读 · 0 评论 -
cuda、torch、torchvision对应版本以及安装
查找torch与torchvision对应版本github链接:https://github.com/pytorch/vision#installation在线下载或者离线下载1、在线下载在pytorch官网选择相应的历史版本,使用conda或者pip安装,使用官网的镜像下载很慢,建议使用其他的镜像源,这里我使用的是阿里的镜像速度还不错。pip install torch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -i https://mirror原创 2021-08-09 10:30:10 · 103856 阅读 · 2 评论 -
mmdetection安装、测试、训练
github:https://github.com/open-mmlab/mmdetection一、配置环境本机的环境是Ubuntu 18.04Cuda 10.2RTX 2080ti在安装过程中遇到了很多的坑,尤其是cuda、pytorch版本的问题。1、创建虚拟环境conda create -n open-mmlab python=3.7 -yconda activate mmdetection2、安装torch环境pip install torch==1.7.1 torchvis原创 2021-08-09 14:44:45 · 1126 阅读 · 1 评论 -
YOLOX安装、测试,自定义coco数据集进行测试
官网链接:https://github.com/Megvii-BaseDetection/YOLOX一、配置环境本机环境是Ubuntu 18.04Cuda 10.2RTX 2080ti1、创建虚拟环境conda create -n yolox python=3.72.安装torchconda activate yoloxpip install torch==1.7.0 torchvision==0.8.0 torchaudio==0.7.03、安装YOLOXgit clone h原创 2021-08-06 16:49:53 · 4130 阅读 · 3 评论