AISciencePlatform / icra2023_synthetic_data_pretraining_for_robotics

icra2023_synthetic_data_pretraining_for_robotics

Requirements

• Install Pytorch

Example using Windows + CUDA11.7 + pip:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

• Install dependencies

pip install -r requirements.txt

Clone this repository

git clone https://github.com/AISciencePlatform/icra2023_synthetic_data_pretraining_for_robotics.git --recursive

• Download the datasets.

• Unzip the file into the folder icra2023_synthetic_data_pretraining_for_robotics

Be sure to have the following structure

Predictions using pre-trained models

• To test the predictions using the pre-trained models run the main.py script.