You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
12 lines
406 B
12 lines
406 B
#!/bin/bash
|
|
|
|
# Python 脚本路径
|
|
PYTHON_SCRIPT="tools/train+sam.py"
|
|
|
|
# 固定的参数
|
|
CONFIG_FILE="configs/vitpose_sam/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTSam_base_coco_256x192.py"
|
|
WORK_DIR="work_dirs/ViTPose_SAM-B_6"
|
|
# RESUME="work_dirs/ViTSam_base_coco_256x192_frozen_sam_ffn/epoch_2.pth"
|
|
|
|
# 执行 Python 脚本并传递固定的参数
|
|
python $PYTHON_SCRIPT $CONFIG_FILE --work-dir $WORK_DIR
|