当前位置: 首页 > news >正文

YOLOX:使用自己数据集训练模型及改进--1.YOLOX环境搭建及运行

YOLOX环境搭建及运行

        YOLO X网络架构是继YOLO v5后,由旷视科技于2021年提出的新一代anthor-free模型,研究者将网络分为输入端、Backbone、PAFPN及Predication,并在Predication提出Decoupled Head、Anchor-free和Multi positives(后文会详细介绍)。

        本篇文章介绍如何通过官方文档下载安装YOLOX模型及环境搭建;如何使用YOLOX训练自己的数据集。 官方文档及论文地址: 

       YOLOX: Exceeding YOLO Series in 2021;

       https://arxiv.org/pdf/2107.08430.pdf

目录-YOLOX环境搭建及运行

一、下载官方YOLOX模型

二、下载安装YOLO X模型及环境搭建

1.前提环境

2.创建虚拟环境

3.下载pytorch(注意pytorch版本需 ≧ 1.7)

4.安装依赖包

5.安装apex

6.安装yolox

7.安装pycocotools

8.下载预训练模型

 9.测试模型是否正常运行

三、YOLOX 训练自己的数据集

1.VOC格式数据集制作

2.COCO数据集制作(VOC格式转COCO格式)

一、下载官方YOLOX模型

1.下载Git(使用Git下载源码,可直接看到源码改动过的地方)

        地址:Git - Downloads;

        安装教程:http://t.csdn.cn/tWyWU

        简易版搭建自己的仓库:

https://www.bilibili.com/video/BV1Ly4y1M7gV/?share_source=copy_web&vd_source=0c3353a32a4f655105ae17a3377691ff

(2)下载YOLO X官方源码

  git clone https://github.com/Megvii-BaseDetection/YOLOX

二、下载安装YOLO X模型及环境搭建

1.前提环境

     Windows10+Python 3.8 +CUDA(根据显卡版本配置)+CUDNN(根据CUDA配置)

2.创建虚拟环境

     (1) 在conda环境下创建YOLOX运行的虚拟环境,本文设置虚拟环境名称为yolox

conda create -n yolox python=3.8

     (2) 激活环境

conda activate yolox 或activate yolox

3.下载pytorch(注意pytorch版本需 ≧ 1.7)

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

   测试是否安装成功(进入python编译环境)

import torch
print(torch.cuda.is_available())  #输出结果为Ture,则安装成功

4.安装依

http://www.lryc.cn/news/486866.html

相关文章:

  • PyTorch使用教程-深度学习框架
  • TON商城与Telegram App:生态融合与去中心化未来的精彩碰撞
  • “乐鑫组件注册表”简介
  • 凹凸/高度贴图、法线贴图、视差贴图、置换贴图异同
  • ZSTD 内存泄漏问题
  • c# npoi操作excel
  • 十二:HTTP错误响应码:理解与应对
  • Rust学习(六):函数式编程
  • 使用 Vue 和 Create-Vue 构建工程化前端项目
  • opencv图片明暗度判断方法
  • QT6学习第三天
  • 计算机网络-MSTP基础实验一(单域多实例)
  • React合成事件及其核心思想详解
  • Datawhale模型减肥秘籍Tasking之模型量化
  • 在云服务器搭建 Docker
  • Redis 的代理类注入失败,连不上 redis
  • 版本控制【Git Bash】【Gitee】
  • Neo4j Desktop 和 Neo4j Community Edition 区别
  • 使用uniapp开发微信小程序使用uni_modules导致主包文件过大,无法发布的解决方法
  • HarmonyOS NEXT应用元服务开发Intents Kit(意图框架服务)事件推荐开发者测试
  • GD32F103 实践-- MCU编译运行
  • SQL复杂数据类型处理
  • ROS第九梯:ROS+VSCode+Python+C++自定义消息发布和订阅
  • 【Linux】指令 + 压缩与解压
  • 力扣(leetcode)题目总结——动态规划篇
  • 数据仓库数据湖湖仓一体解决方案
  • 微信小程序 最新获取用户头像以及用户名
  • 无人机在森林中的应用!
  • Seatunnel解决Excel中无法将数字类型转换成字符串类型以及源码打包
  • 在阿里云快速启动Appsmith搭建前端页面