构建no-text数据集

This commit is contained in:
2026-02-20 14:13:25 +08:00
parent ee257bcb6c
commit d85cce8a52
5 changed files with 194 additions and 62 deletions

25
build_no_text_dataset.sh Executable file
View File

@@ -0,0 +1,25 @@
SEG_ROOT="/home/cyx6123/DuodenoVLA/data/ACT/aloha/act/data/raw_data/00-follow"
OUT_DIR="/home/cyx6123/DuodenoVLA/data/ACT/aloha/act/data/follow-no-text"
SCRIPT="/home/cyx6123/DuodenoVLA/data/ACT/aloha/act/build_endoscope_act_dataset.py"
mkdir -p "$OUT_DIR"
i=50
for d in "$SEG_ROOT"/follow_seg_*; do
[ -d "$d" ] || continue
echo "Building $d -> episode_$i"
python "$SCRIPT" \
--segment_dir "$d" \
--output_dir "$OUT_DIR" \
--episode_idx "$i" \
--max_frames -1 \
--camera_name top \
--crop 733 30 1754 1051 \
--resize 224 224 \
--motion_window 3 \
--motion_threshold 0.05 \
--state_norm minus1_1 \
--action_norm minus1_1 \
--trim_stationary_edges \
--no_text_instruction
i=$((i+1))
done