使用mmdetection来训练自己的数据集(visdrone)(一)

ops/2024/9/23 7:00:36/

环境:

创建虚拟环境

conda create -n mmdet python=3.8 -y

安装torch
https://pytorch.org/get-started/previous-versions/

pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
torch                  2.1.0+cu121
torchaudio             2.1.0+cu121
torchvision            0.16.0+cu121

使用 MIM 安装 MMEngine 和 MMCV。

pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.0"

但是我们torch版本是12.1 不是很兼容,通过下面的来更改mmcv版本
https://mmcv.readthedocs.io/en/latest/get_started/installation.html

安装 mmdetection

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -v -e .
# "-v" 指详细说明,或更多的输出
# "-e" 表示在可编辑模式下安装项目,因此对代码所做的任何本地修改都会生效,从而无需重新安装。

为了验证 MMDetection 是否安装正确,我们提供了一些示例代码来执行模型推理。

  1. 我们需要下载配置文件和模型权重文件。
mim download mmdet --config rtmdet_tiny_8xb32-300e_coco --dest .
  1. 如果你通过源码安装的 MMDetection,那么直接运行以下命令进行验证:
python demo/image_demo.py demo/demo.jpg rtmdet_tiny_8xb32-300e_coco.py --weights rtmdet_tiny_8xb32-300e_coco_20220902_112414-78e30dcc.pth --device cpu

结果图
请添加图片描述


http://www.ppmy.cn/ops/22915.html

相关文章

OpenHarmony语言基础类库【@ohos.util.HashMap (非线性容器HashMap)】

HashMap底层使用数组链表红黑树的方式实现,查询、插入和删除的效率都很高。HashMap存储内容基于key-value的键值对映射,不能有重复的key,且一个key只能对应一个value。 HashMap和[TreeMap]相比,HashMap依据键的hashCode存取数据&…

Midjourney与waifu2x双剑合璧:完美打造超高清动漫图像

在追求完美的动漫图像时,质量和分辨率是两个关键因素。Midjourney(一个神秘而强大的AI图像生成工具)与waifu2x(一个专门用于放大动漫风格图像的AI工具)的结合使得创造超高清的动漫图像变得触手可及。本文将引导您如何使…

2024.4.28力扣每日一题——负二进制转换

2024.4.28 题目来源我的题解方法一 进制转换方法二 模拟进位 题目来源 力扣每日一题;题序:1017 我的题解 方法一 进制转换 对于以-2为基数的系统,可以这样理解:在-2进制中,每一位的权重是-2的幂。这与传统的二进制表…

Web前端开发之CSS_1

CSS选择器字体属性背景属性文本属性表格属性 1. CSS 1.1 CSS简介 CSS(Cascading Style Sheets)层叠样式表,又叫级联样式表,简称样式表。CSS文件后缀名为 .css 。CSS用于HTML文档中元素样式的定义。使用CSS可以让网页具有美观一致…

linux运行jar

安装Java环境 下载Java安装包 首先,需要从Oracle官网或其他可信渠道下载适用于Linux系统的Java安装包。根据系统架构(32位或64位)选择合适的安装包。 安装Java 下载完成后,通过终端进入安装包所在目录,使用tar命令…

第三部分 Vue讲解(22-25)(代码版)

22.自定义指令 22.1 自定义指令基本语法 22.1.2 自定义指令.vue <template><div id"app"><!-- v-model相当于:value"XXX" input"msgXXX"--><input v-model"msg" type"text"><input :value…

Mac环境安装任意版本的node

背景 由于在使用node的时候需要频繁的切换node的版本。在windows上可以通过安装不同版本的软件进行实现&#xff0c;在mac上可以更加方便快捷的实现这个过程哦&#xff01; 全局安装n npm install -g n选择不同的版本 # 安装最新稳定版 node n stable # 安装最新版本 node …

金属冶炼及压延加工制造数字孪生可视化平台,推进行业数字化转型

金属冶炼及压延加工制造数字孪生可视化平台&#xff0c;推进行业数字化转型。随着科技的不断进步和工业的快速发展&#xff0c;金属冶炼及压延加工行业正面临着前所未有的挑战和机遇&#xff0c;数字化转型成为了行业发展的必然趋势。在这个过程中&#xff0c;数字孪生可视化平…