【mmdetection实践】（三）使用华为云安装训练mmdetection

最新推荐文章于 2024-05-15 17:13:00 发布

原创

最新推荐文章于 2024-05-15 17:13:00 发布 · 1.2k 阅读

3 ·

CC 4.0 BY-SA版权

本文介绍了如何在华为云环境中安装和训练mmdetection。首先，将mmdetection上传至OBS以供训练作业使用，而不是直接从GitHub下载。接着，创建新的启动文件main.py，置于mmdetection目录下，而不是修改原有的train.py和test.py。最后，针对已有的backbone权重，修改配置文件faster_rcnn_r50_fpn_1x.py。

mmdetection传到OBS中

我们需要在训练作业中安装mmdetection。有两种方法，一种是存在OBS中，然后传入训练作业中，另外一种使用代码从github上下载。这里，我们使用第一种方法，因为这样子，我门还可以对mmdetection进行修改。所以，我们先下载mmdetection，然后上传到OBS中。并将此目录作为代码目录。

写启动代码

mmdetection内包括train.py和test.py的启动代码。这里我们不选择修改这两个py，而是新写一个main.py，作为启动代码，将其放在mmdetection的目录下。

import moxing as mox
import os

# 接受训练作业传入的两个参数
import argparse
parser = argparse.ArgumentParser(description='training process')
parser.add_argument('--train_url', type=str, default='')  # 训练作业的输出位置
parser.add_argument('--data_url', type=str, default='')  # data在obs上的地址
args, _ = parser.parse_known_args()