btx0424 / GenDiversity

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GenDiversity

Installation

git clone --recursive git@github.com:btx0424/GenDiversity.git
# install GenSim
cd projects/GenSim && pip install -e .
# install diffusion_policy
cd ../diffusion_policy && pip install -e .
# install this repo
cd .. && pip install -e .

Data Collection

For GenSim:

# at gen_diversity/scripts
python dataset/collect_gensim.py n=100 task=packing-boxes-pairs-seen-colors

Training

DP for GenSim:

# at diffusion_policy/
bash run.sh

About


Languages

Language:Python 96.7%Language:Shell 3.3%