mmdet整合train和test

最新推荐文章于 2024-08-08 08:01:22 发布

carry_hjr

最新推荐文章于 2024-08-08 08:01:22 发布

阅读量494

点赞数 1

分类专栏：遥感图像处理与深度学习

本文链接：https://blog.csdn.net/carry_hjr/article/details/117415323

版权

遥感图像处理与深度学习专栏收录该内容

22 篇文章 16 订阅

订阅专栏

需求

train完自动test

实现

shell里面解析一下config做成checkpoint

#!/usr/bin/env bash

PYTHON=${PYTHON:-"python"}

CONFIG=$1
GPUS=$2
PORT=${PORT:-29500}

$PYTHON -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
    $(dirname "$0")/train.py $CONFIG --launcher pytorch ${@:3}

tmp=($(echo "$CONFIG" | tr '/' '\n'))
CHECKPOINT="work_dirs/${tmp[1]}/${tmp[2]/.py/}/latest.pth"

PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
    $(dirname "$0")/test.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}