单目3D初始代码
This commit is contained in:
21
tools/model_merging/export_single_roi_yolo26.sh
Executable file
21
tools/model_merging/export_single_roi_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 tools/model_merging/export_single_roi_yolo26.py \
|
||||
--model-path runs/detect/train_mono3d_roi0_202603291330_epoch61.pt \
|
||||
--save-dir runs/export/train_mono3d_single_roi0_202603291330-postprocessed-original \
|
||||
--imgsz 768 352 \
|
||||
--opset 11 \
|
||||
--postprocessed-outputs
|
||||
# --detections-only
|
||||
|
||||
# Example: export ROI1 instead
|
||||
# python tools/model_merging/export_single_roi_yolo26.py \
|
||||
# --model-path runs/detect/train_mono3d_roi1_202603291330_epoch99.pt \
|
||||
# --save-dir runs/export/train_mono3d_single_roi1_202603291330-detections \
|
||||
# --imgsz 768 352 \
|
||||
# --opset 15 \
|
||||
# --detections-only
|
||||
Reference in New Issue
Block a user