深度学习openMMLab的介绍和使用
文章目录
- MMCV介绍
- MMCV的安装
- 修改链接中的cu113
- 修改链接中的torch1.10.0
- 物体分类MMCLS
- 源码下载
- 配置参数解读
- 配置文件的组成
- 如何生成完整配置文件
- 定义自己的数据集
- 构建自己的数据集
- 训练自己的任务
- 物体检测MMDetection
- 语义分割MMSegmentation
- 姿态估计MMPose
- 未完成,持续更新中。。。
立于皓月之边,不弱星光之势
MMCV介绍
MMCV 开源框架从属于 OpenMMLab 系列,OpenMMLab 系列是专为计算机视觉不同方向建立统一而开放的代码库。目前已经开源了多个算法框架,例如物体检测的 MMDetection、语义分割的 MMSegmentation 和姿态估计的 MMPose 等等。
MMCV的安装
安装地址:https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
修改链接中的cu113
这里需要修改cu113为自己的cuda版本
查看自己cuda版本的方法:打开命令行输入nvcc -V
我这里是11.7,则改为117。如果没有则可以找相邻版本。
修改链接中的torch1.10.0
这里需要修改torch1.10.0为自己的torch版本
-
查看自己torch版本的方法:打开命令行输入
conda env list
查看所有环境。 -
-
输入
activate pytorch1.13
进入虚拟环境 -
-
输入
pip list
查看torch版本 -
下载之后打开命令提示符先切入到虚拟环境再切换到下载目录
输入命令:pip install mmcv_full-1.7.1-cp39-cp39-win_amd64.whl
物体分类MMCLS
源码下载
https://github.com/open-mmlab/mmclassification