【目标跟踪】【MMTracking的部署与开发】01 安装与部署

1.准备

• Linux | macOS | Windows

• Python 3.6+

• PyTorch 1.3+

• CUDA 9.2+ (If you build PyTorch from source, CUDA 9.0 is also compatible)

• GCC 5+

• MMCV

• MMDetection

兼容的MMTracking、MMCV和MMDetection版本如下所示。请安装正确的版本,以避免出现安装问题。

MMCV:
MMCV

MMDetection:
MMDetection:

2.安装

2.1 安装细节

官方文档上有个推荐的教程,我参考这个教程,结合网上其他的教程,自己在Linux服务器上面成果部署了一套环境

1.创建虚拟环境并激活

python 复制代码
conda create -n open-mmlab python=3.9 -y
conda activate open-mmlab

2.安装Pytorch

注意要安装与自己电脑CUDA版本对应的Pytorch

对于CUDA11.X版本的 我建立安装Pytorch的1.8版本或者1.7

如果CUDA是10.X版本的,官方文档上建议安装1.5版本的Pytorch

python 复制代码
# CUDA 11.3
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

上面的代码是去Pytorch官网上找的,

找到官网首页

添加链接描述

点击Previous versions of Pytorch

就可以根据自己操作系统和CUDA的版本情况来选择对应的Pytorch的安装指令

3.安装VOT视频目标跟踪评估需要的依赖库(这个是可选的)

如果您需要评估VOT挑战,请在安装mmcv和mmdetection之前安装虚拟工具包,以避免某些依赖项之间可能存在的numpy版本需求冲突

python 复制代码
pip install git+https://github.com/votchallenge/toolkit.git

4.安装mmcv-full

推荐采用以下的安装方式:

python 复制代码
# pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html

cu_version是CUDA的版本,torch_version是Pytorch的版本

比如我的CUDA的版本是11.3,我的Pytorch的版本是1.10

我的mmcv-full的安装指定就是:

python 复制代码
pip install mmcv-full==1.7.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

在这里我指定了mmcv-full的安装版本是1.7.0,我推荐使用1.5.0-2.0.0之间的版本

5.安装MMDetection

有两种安装方法:

一种是直接安装,或者先安装mim

python 复制代码
pip install openmim
mim install mmdet==2.25.1

2.25.1是mmdetection的版本号,这个mmcv mmdetection和mmtracking之间都存在着版本兼容关系,建议大家查一下。

另外一种就是git clone

python 复制代码
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .
# or "python setup.py develop"

6.git clone MMtracking

python 复制代码
git clone https://github.com/open-mmlab/mmtracking.git
cd mmtracking

7.安装MMtracking

python 复制代码
pip install -r requirements/build.txt
pip install -v -e .
# or "python setup.py develop"

8.安装额外的依赖库

对于MOTChallenge任务

python 复制代码
pip install git+https://github.com/JonathonLuiten/TrackEval.git

对于LVIS evaluation

python 复制代码
pip install git+https://github.com/lvis-dataset/lvis-api.git

对于TAO evaluation:

python 复制代码
pip install git+https://github.com/TAO-Dataset/tao.git

需要说明的是,按照上述说明,MMTracking安装在开发模式下,对代码进行的任何本地修改都将生效,无需重新安装。

也就是说这样安装的话可以直接在上面做自己的修改

相关推荐
曦月逸霜2 小时前
啥是RAG 它能干什么?
人工智能·python·机器学习
AI医影跨模态组学2 小时前
Lancet Digit Health(IF=24.1)广东省人民医院刘再毅&南方医科大学南方医院梁莉等团队:基于可解释深度学习模型预测胶质瘤分子改变
人工智能·深度学习·论文·医学·医学影像·影像组学
应用市场2 小时前
AI 编程助手三强争霸(2026 版):Claude、Gemini、GPT 各自擅长什么?
人工智能·gpt
CSND7402 小时前
YOLO resume断点续训(不能用官方的权重,是自己训练一半生成的last.pt)
深度学习·yolo·机器学习
AC赳赳老秦2 小时前
供应链专员提效:OpenClaw自动跟踪物流信息、更新库存数据,异常自动提醒
java·大数据·服务器·数据库·人工智能·自动化·openclaw
脑极体2 小时前
从Token消耗到DAA增长,AI价值标尺正在重构
人工智能·重构
csdn小瓯2 小时前
LangGraph自适应工作流路由机制:从关键词匹配到智能决策的完整实现
人工智能·fastapi·langgraph
QYR-分析2 小时前
高功率飞秒激光器行业发展现状、市场机遇及未来趋势分析
大数据·人工智能
AI医影跨模态组学3 小时前
J Clin Oncol(IF=43.4)美国Cedars-Sinai医学中心等团队:基于计算组织学人工智能的晚期胰腺癌化疗选择预测性生物标志物的开发与验证
人工智能·机器学习·论文·医学·医学影像·影像组学
冬奇Lab3 小时前
RAG 系列(十六):Graph RAG——用知识图谱解决多跳关系问题
人工智能·llm