31 lines
807 B
Bash
31 lines
807 B
Bash
#!/usr/bin/env bash
|
|
|
|
PORT=${PORT:-29500}
|
|
CFGFILE=tools/demo_yolox_cfg.py
|
|
|
|
CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch \
|
|
--nproc_per_node=2 \
|
|
--master_port=$PORT \
|
|
mmdetection/tools/train.py $CFGFILE \
|
|
--launcher pytorch ${@:3}
|
|
|
|
CFGFILEBNAME=$(basename $CFGFILE)
|
|
PTHFILE=$(find work_dirs/${CFGFILEBNAME%.*} -name "best*.pth" -type f | head -1)
|
|
ONNXFILE=${PTHFILE%.pth}.onnx
|
|
KNERONOUTJSONFILE=${CFGFILE%.py}_kneron_output_eval.json
|
|
|
|
python mmdetection/tools/deployment/pytorch2onnx_kneron.py \
|
|
$CFGFILE \
|
|
$PTHFILE \
|
|
--output-file $ONNXFILE \
|
|
--skip-postprocess \
|
|
--verify
|
|
|
|
python tools/parse_cfg_onnx_to_dag.py $CFGFILE $PTHFILE
|
|
|
|
python mmdetection/tools/test_kneron.py \
|
|
$CFGFILE \
|
|
$ONNXFILE \
|
|
--out-kneron $KNERONOUTJSONFILE \
|
|
--eval bbox
|