Skip to content

Commit fee1d80

Browse files
author
Your Name
committed
Enhance droid_combined_ingestion.py to improve trajectory generation by adding checks for TFDS data availability and ensuring valid trajectory data before file creation. Update camera serial mapping logic and adjust dataset loading to use 'droid_100'. Modify .gitignore to include 'droid_100' directory.
1 parent 8896032 commit fee1d80

6 files changed

Lines changed: 2079 additions & 10 deletions

File tree

examples/droid/.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,3 +6,4 @@ f1_matrix_results/
66
trajectory_captioning_results/
77
droid_combined_data/
88
huggingface_cache/
9+
droid_100/

examples/droid/Dockerfile

Lines changed: 43 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,43 @@
1+
# docker build --network=host -t droid-downloader .
2+
# docker run --network=host -v $(pwd)/droid_data:/root/droid-example/droid_downloaded_data droid-downloader bash -c "python3 droid_downloader.py"
3+
FROM stereolabs/zed:4.2-runtime-cuda11.8-ubuntu22.04
4+
5+
# RUN apt-get update -y && apt-get install -y \
6+
# fish \
7+
# python3-pip \
8+
# python3-opencv \
9+
# git
10+
11+
# Install Python dependencies
12+
RUN pip install \
13+
argparse \
14+
scipy==1.10.1 \
15+
h5py \
16+
gcsfs \
17+
tensorflow_datasets \
18+
tensorflow \
19+
ray[default] \
20+
flask \
21+
spacy \
22+
numpy \
23+
requests \
24+
opencv-python
25+
26+
# Install Google Cloud SDK
27+
RUN curl -O https://dl.google.com/dl/cloudsdk/channels/rapid/install_google_cloud_sdk.bash
28+
RUN chmod +x install_google_cloud_sdk.bash
29+
RUN ./install_google_cloud_sdk.bash --disable-prompts
30+
31+
# Add gsutil to PATH
32+
ENV PATH="/root/google-cloud-sdk/bin:$PATH"
33+
34+
# Set working directory
35+
WORKDIR /root/droid-example
36+
37+
# Copy the scripts
38+
COPY . .
39+
40+
# Install robodm (assuming it's in the parent directory during build)
41+
# You'll need to mount or copy the robodm package
42+
# COPY ../../robodm /root/robodm_pkg
43+
# RUN cd /root/robodm_pkg && pip install -e .

0 commit comments

Comments
 (0)