mmdet整合train和test

需求

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}