单目3D初始代码

This commit is contained in:
zhao.zhu
2026-06-24 09:35:46 +08:00
commit 04a5895b6b
1153 changed files with 340700 additions and 0 deletions

10
train_mono3d_single_node.sh Executable file
View File

@@ -0,0 +1,10 @@
python -m torch.distributed.run --master_port 29501 --nproc_per_node 8 \
train_mono3d.py --device 0,1,2,3,4,5,6,7 \
--epochs 100 --lr0 0.001 --batch 128 --imgsz 768,352 \
--roi roi0 \
--pretrained runs/detect/train_mono3d_roi0_202603251430/weights/best.pt --roi_metrics_only \
--warmup_epochs 10 \
--loss_3d_warmup_epochs 10 \
--loss_3d_ramp_epochs 20 \
--loss_3d_weight_max 0.1 \
--exp_dir=./runs/detect/train_mono3d_roi0_20260327