Project Description

Pocket Racer is a platform designed for simulating and training multi-agent autonomous racing models. Leveraging advanced transformer architectures, this platform enables the study and improvement of autonomous racing strategies in educational situations.

Training Instructions

Training an autonomous model can be quite challenging. Please follow these instructions carefully to ensure successful training:

Installation

To set up the Pocket Racer platform, please follow these steps:


                  # Clone the repository
                  git clone https://github.com/yourgithub/pocket-racer.git
                  
                  # Navigate to the project directory
                  cd pocket-racer
                  
                  # Install required dependencies
                  pip install -r requirements.txt
                                  

Data Processing

To use the data_processor.py script for preparing and processing your datasets, run the following command:


python data_processor.py --n_stacked <number_of_stacked_images> --img_path <path_to_images> --csv_path <path_to_csv> --w <width> --h <height> --d <depth> --concatenate <concatenate_flag> --prediction_mode <mode>


Modify the parameters according to your dataset's requirements:

  1. n_stacked: Number of images to stack for input data.
  2. img_path: Path to the directory containing image files.
  3. csv_path: Path to the CSV file containing metadata or labels.
  4. w, h, d: Width, height, and depth of the processed images.
  5. concatenate: Boolean flag to determine whether to concatenate data.
  6. prediction_mode: Specifies the mode of prediction, e.g., 'linear' or 'categorical'.

Training

To train the model using the provided scripts, you can run:


python train_script.py --epochs 50 --batch_size 1000


Modify the parameters as needed for your specific training setup.

Features

Our platform offers the following key features:

  1. Model Training: Train vision transformers tailored for steering prediction.
  2. Performance Metrics: Evaluate model performance with metrics such as MAE, RMSE, and R².
  3. Customizable Training Options: Configure training epochs, batch size, and image preprocessing through command-line arguments.

For detailed code, please visit our GitHub repository

Contributing

Contributions to Pocket Racer are welcome! Please consult the CONTRIBUTING.md file for guidelines on how to make contributions.