首页 > 其他分享 >手把手带你用RT-DETR模型部署训练自己的数据集

手把手带你用RT-DETR模型部署训练自己的数据集

时间:2024-03-27 10:32:34浏览次数:29  
标签:RT img format 手把手 yolo json context path DETR

本文主要介绍如何部署、训练RT-DETR模型,将从两种方式进行介绍:一种是RTDETR自带的项目文件,另一种是ultralytics仓库下将训练的模型改为rtdetr的基本模型。

1.配置环境

1.1新建虚拟环境

我们可以在电脑上新建一个虚拟环境

conda create -n RT-DETR python=3.10
激活环境
conda activate RT-DETR

1.2安装pytorch 

 然后安装pytorch,可以去pytorch官网找自己电脑的cuda版本对应的命令直接运行

首先我们使用以下命令查看自己的电脑cuda版本

nvidia-smi

比如说下面是我的cuda版本,如图所示cuda版本为12.3

 然后我们去pytorch官网找自己cuda对应版本的ptyorch,如下是pytorch官网 

Previous PyTorch Versions | PyTorch

 我这里选择的是上图这条命令,找版本的时候只要小于电脑的cuda版本即可,将上图命令复制到项目终端运行,pytorch就安装好了。

2.自带项目文件运行

首先我们去下载官方的RT-DETR项目文件

RT,img,format,手把手,yolo,json,context,path,DETR
From: https://blog.csdn.net/m0_61751664/article/details/137065199

相关文章

  • Rancher Dashboard 无法访问 引申发现K8S报错Unable to connect to the server: x509:
    访问RancherDashboard,发现无法访问由于笔者的rancher是用docker部署的,查看rancherlog:dockerlogs[container-name]截取一部分报错如下:..2024-03-2406:52:27.085313I|embed:readytoserveclientrequests2024-03-2406:52:27.085567I|etcdserver:published{N......
  • 手把手教你做阅读理解题-初中中考阅读理解解题技巧002
    PDF格式公众号回复关键字:ZKYD002阅读理解技巧,在帮助读者有效获取和理解文本信息方面发挥着重要作用,熟练掌握如下6个技巧,可快速突破阅读理解1预览文章结构在开始深入阅读之前,快速浏览文章的标题、段落开头和结尾,可以迅速把握文章的主题、大致内容和结构标题通常能概括文章......
  • How-To-Ask-Questions-The-Smart-Way
    文章转载至github如侵立删https://github.com/ryanhanwu/How-To-Ask-Questions-The-Smart-Way提问的智慧HowToAskQuestionsTheSmartWayCopyright©2001,2006,2014EricS.Raymond,RickMoen本指南英文版版权为EricS.Raymond,RickMoen所有。原文网址:http://......
  • Devvortex Hack the box
    nmap-A-oNinitial_scan10.10.11.242发现打开了80和22用dirsearch和fuff扫子域名和子目录直接发现目录/administrator进去发现是Joomla页面发现版本是4,那么就和漏洞Joomla(CVE-2023-23752)一样可以看https://xz.aliyun.com/t/12175?time__1311=mqmhD5DK7IejhDBdPx2DUEF......
  • FPGA接口系列——UART
    FPGA接口系列——UART一、UART简介UART是一种采用异步串行通信方式的通用异步收发传输器。这里我们主要弄明白两个问题:①什么是串行通信,与并行通信有什么区别②同步串行通信和异步串行通信有什么区别③我们常说的UART和RS232以及RS485又有什么区别?串行通信和并行通信......
  • std::sort 错误"Expression : invalid operator <"
    解决:std::sort的比较函数,切记仅使用小于或大于,不要使用小于等于或大于等于。即所谓的“strictweakordering”,也就是说,如果a==b,则返回的应该是false,如果返回的是true,则会出上面的错这个问题是标准库sort实现导致的参考https://blog.csdn.net/qq_35097289/article/details/1046......
  • 1.VirtualBox虚拟机软件的安装
    1.1下载安装VirtualBox进入VirtualBox官网下载安装(www.virtualbox.org/wiki/Downloads)1.2 设置VirtualBox默认存储文件夹。点击管理>>全局设定>>默认虚拟电脑位置,把默认位置修改为你想要放的位置。 1.3在VirtualBox创建虚拟机创建虚拟机,单击新建,机器名称:Hadoop,类......
  • Windows Packet Divert(WinDivert)是一个适用于Windows 10、Windows 11和Windows Server
    WindowsPacketDivert(WinDivert)是一个适用于Windows10、Windows11和WindowsServer的用户模式数据包捕获和重定向工具。WinDivert允许用户模式应用程序捕获/修改/丢弃发送到/从Windows网络堆栈的网络数据包。总之,WinDivert可以:捕获网络数据包过滤/丢弃网络数据包嗅探......
  • C#JsonConvert.DeserializeObject反序列化与JsonConvert.SerializeObject序列化
    原文链接:https://blog.csdn.net/qq_45451847/article/details/120434797JSONJSON序列化是将对象转换为JSON格式的字符串,而JSON反序列化是将JSON格式的字符串转换为对象。对于JSON大家都了解,JSON是一种轻量级的文本数据交换格式而非编程语言,既然是数据交换格式,那就需要不断的进......
  • 使用TensorRT-LLM部署ChatGLM3
    说明:TensorRT-LLM是NVIDIA开发的高性能推理TensorRT-LLM是NVIDIA推出的一个专门用于编译和优化大语言模型推理的综合程序库。可在NVIDIAGPU上加速和优化最新的大语言模型(LLM)的推理性能。官网地址为:https://github.com/THUDM/ChatGLM3/blob/main/tensorrt_llm_demo/READ......