The repository is a fork of the original project by xmba15 and provides an adapted workaround for a training with data from the Waymo Open Dataset. A training/val and testing set can be provided on demand.
- tested on the following environment:
- OS: tested on Ubuntu 20.04
- Tensorflow 2.6.0
- python 3.8
- installing the dependencies:
- python3 -m pip install -r requirements.txt
- Sample of automatically generated ground truth from LiDAR pointclouds
- Scalability in Perception for Autonomous Driving: Waymo Open Dataset
After downloading the dataset, run
python3 ./train.py
model weights will be saved into ./saved_models directory
- Download pretrained model weights with
t.b.d.
- Test on an image
python3 ./test_single_image.py --model_path [path/to/model/weights]