pytorch
文章平均质量分 65
GY—Monkey
这个作者很懒,什么都没留下…
展开
-
mmdetection训练(1)voc格式的数据集(自制)
(1)在代码根目录新建myconfig.py的文件,(2)复制以下内容到其中:新的配置文件主要是分为三个部分1、倒入相应的库文件(base2、模型加载文件:一定要家在修改num_classses=‘你的类别’3、数据集配置:直接复制configs/base/datasets/voc0712.py即可# 新配置继承了基本配置,并做了必要的修改# 我们还需要更改 head 中的 num_classes 以匹配数据集中的类别数。原创 2024-05-15 17:26:10 · 583 阅读 · 0 评论 -
【无标题】ubuntu环境手把手配置mmdetection
官网路径:https://mmdetection.readthedocs.io/zh-cn/latest/get_started.html。原创 2024-05-14 09:10:12 · 368 阅读 · 0 评论 -
ubuntu20.04中安装配置docker nvidia容器来实现宿主机GPU的调用
7、当一个新的 Docker 发布时,你可以使用标准的sudo apt update && sudo apt upgrade流程来升级 Docker 软件包。我们将会启用 Docker 软件源,导入 GPG key,并且安装软件包。5、想要安装 Docker 最新版本,运行下面的命令。4、现在,Docker 软件源被启用了,你可以安装软件源中任何可用的 Docker 版本。6、一旦安装完成,Docker 服务将会自动启动。(这里指推荐使用官方的安装方法,snap的方法不推荐,各种命令不是很通用)原创 2024-03-11 16:03:33 · 2127 阅读 · 2 评论 -
使用mmrotate训练 Dronevehicle 数据集
目前看到有很多教程讲述mmrotate训练自制数据集或者dota数据集,这里讲一下在visitdrone数据集上的训练过程。原创 2023-04-07 08:58:21 · 2749 阅读 · 13 评论 -
mmdetection 环境配置与简单测试(mmrotate同理)
其中,_base_中datasets是定义了数据集加载的格式(coco或者voc)与数据的路径。configs中的改进模型都继承于这几个基本模型框架,通过设置不同的层数和参数建立的新模型。这里用的win11进行的测试,后续也会补充Ubuntu的配置方法,几乎同理。这里采用的是基于conda的安装,30系列显卡要用cuda11以上的环境。core中存放了进行检测的主要机理过程,并封装函数,核心的计算原理过程。1、首先要安装pytorch,查看官网的教程的链接:pytorch。4、验证是否安装成功。原创 2023-04-04 17:12:15 · 931 阅读 · 0 评论 -
pytorch 网络参数显存计算
一、使用torchstatpip install torchstat from torchstat import statimport torchvision.models as modelsmodel = models.resnet152()stat(model, (3, 224, 224))关于stat函数的参数,第一个应该是模型,第二个则是输入尺寸,3为通道数。我没有调研该函数的详细参数,也不知道为什么使用的时候并不提示相应的参数。二、使用torchsummarypip inst原创 2022-05-03 15:54:35 · 612 阅读 · 0 评论 -
常见卷积尺寸计算
在搭配深度学习多个卷积层时我们经常要计算卷积层的输出张量的尺寸大小,可以用如下公式计算:1, 公式卷积层输出尺寸: o = ?(i + 2p - k) / s? + 1式中,i:输入尺寸;o:输出尺寸;p:padding;k: kernel_size;s: stride。?…?表示向下取整。2, 推导过程这个公式不用死记,下面我用非常便于理解的方法描述这个推导。卷积就是对相邻的一片数据进行加权求和得到一个数的一种“合并”操作,将此操作对输入张量进行滑动扫描以得到输出张量。循着这个过程,我们很容易推原创 2022-04-15 11:23:39 · 9974 阅读 · 0 评论 -
使用ps制作大型图片的分割数据集(遥感,sar)
文章目录前言一、photoshop二、使用步骤1.引入库2.读入数据总结前言使用传统的标注软件,labelme与百度ai智能分割软件eiseg,都无法打开遥感,sar图像的原图,会直接爆存储闪退,故出此教程来进行数据集的制作。一、photoshopps的强大二、使用步骤1.引入库代码如下(示例):import numpy as npimport pandas as pdimport matplotlib.pyplot as pltimport seaborn as snsimp原创 2022-03-01 10:50:47 · 1050 阅读 · 0 评论 -
【学习语法】————————chunk()和squeeze() and unsqueeze()用法
语法函数学习学习一、chunk()二、squeeze()与unsqueeze()学习通俗易懂理解函数的用法以及执行的机理一、chunk()其中chunk(a,b) 有两个参数输入,分别为a,与ba代表要分的块数b = 0时是横向分割b = 1时是纵向分割(我个人记不住上述这个,老记混)b=0时,是对第一维分割(例:2x2 使用chunk(2,0)——>1x2,1x2)b = 1 时,是对第二维分割(例:例:2x2 使用chunk(2,1)——>2x1,2x1)i原创 2021-11-29 16:07:22 · 955 阅读 · 0 评论