#!/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