You can deploy this model and start earning money today!

mmkuznecov/
SynthOccPredModels

voxel_car

occupancy-prediction

reinforcement-learning

Views

No views yet

SynthOccPred / voxel_car models

This repository contains two trained artifacts for the voxel_car project:

Artifact	Hub path
Occupancy predictor checkpoint	`occupancy/ckpt_best.pt`
PPO policy for RL planner	`rl/ppo_voxel_car_final.zip`

Use from Python

python

from voxel_car.hub import load_occnet_from_hf, load_ppo_from_hf

model, ego_cfg, cam_cfg, image_hw, device = load_occnet_from_hf(
    repo_id="mmkuznecov/SynthOccPredModels",
)

ppo = load_ppo_from_hf(
    repo_id="mmkuznecov/SynthOccPredModels",
)

Use in the Gradio demo

bash

export VOXEL_CAR_HF_REPO="mmkuznecov/SynthOccPredModels"
python app.py

The demo can run closed-loop scenarios with either:

standard A* planner over OccNet occupancy predictions
PPO-RL planner over OccNet occupancy predictions