22 lines
797 B
Bash
22 lines
797 B
Bash
|
#!/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
|