22 lines
779 B
Bash
22 lines
779 B
Bash
|
#!/bin/bash
|
||
|
|
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
PROJECT_ROOT="$(cd "${SCRIPT_DIR}/../.." && pwd)"
|
||
|
|
export PYTHONPATH="${PROJECT_ROOT}:${PYTHONPATH}"
|
||
|
|
PYTHON_BIN="/deeplearning_team/ydong/dongying/miniconda/envs/pytorch1.9/bin/python"
|
||
|
|
|
||
|
|
if [ ! -x "${PYTHON_BIN}" ]; then
|
||
|
|
echo "ERROR: pytorch1.9 python not found: ${PYTHON_BIN}"
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
"${PYTHON_BIN}" tools/model_merging/merge_models_of_2roi_yolo26.py \
|
||
|
|
--roi0-model-path runs/detect/mono3d_roi0_20260506_epoch99.pt \
|
||
|
|
--roi1-model-path runs/detect/mono3d_roi1_20260506_epoch99.pt \
|
||
|
|
--save-dir runs/export/train_mono3d_two_roi_20260506-drop_fake_3d_branch \
|
||
|
|
--imgsz 768 352 \
|
||
|
|
--opset 11 \
|
||
|
|
--edge-head-mode drop \
|
||
|
|
--fake-3d-branch-mode drop
|
||
|
|
# --postprocessed-outputs
|