构建no-text数据集
This commit is contained in:
25
build_no_text_dataset.sh
Executable file
25
build_no_text_dataset.sh
Executable 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
|
||||
Reference in New Issue
Block a user