单目3D初始代码
This commit is contained in:
21
tools/model_merging/merge_models_of_2roi_yolo26.sh
Executable file
21
tools/model_merging/merge_models_of_2roi_yolo26.sh
Executable file
@@ -0,0 +1,21 @@
|
||||
#!/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
|
||||
Reference in New Issue
Block a user