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.
15 lines
534 B
15 lines
534 B
#!/bin/bash
|
|
|
|
# Python 脚本路径
|
|
PYTHON_SCRIPT="demo/top_down_img_demo.py"
|
|
|
|
# 固定的参数
|
|
CONFIG_FILE="configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_huge_coco_256x192.py"
|
|
CHECKPOINT="checkpoints/vitpose/vitpose-h.pth"
|
|
IMG_ROOT="tests/data/hengdian"
|
|
JSON_FILE="tests/data/hengdian/test_hengdian.json"
|
|
OUT_IMG_ROOT="test/hengdian/test"
|
|
|
|
# 执行 Python 脚本并传递固定的参数
|
|
python $PYTHON_SCRIPT $CONFIG_FILE $CHECKPOINT --img-root $IMG_ROOT \
|
|
--json-file $JSON_FILE --out-img-root $OUT_IMG_ROOT
|