python3 loop_over_dataset_ID_S1_EX1.py
python3 loop_over_dataset_ID_S1_EX2.py
Find 10 examples of vehicles with varying degrees of visibility in the point-cloud
I found 15 vehicles with different degrees of visibility. Their visibilities depends on 3 factors:
- The yaw angle heading toward the lidar sensor.
- The distance from the vehicle to the lidar sensor.
- If the car is behind other objects.
Cars 1, 2, 3, 11, 12, 14, 15 are represented by fewer lidar lines because they are far from the lidar sensor.
Cars 4, 5, 6, 7, 8, 9, 10, 13 are represented by more lidar lines because they are nearer the lidar sensor.
Cars 10, 11, 12 are partially occluded by the cars in front of them. That means lasers bounce in the nearer cars and cannot reach the cars behind. If you carefully observe, behind every car, there is an empty space where lasers cannot reach.
Try to identify vehicle features that appear stable in most of the inspected examples and describe them
The yaw angles of cars 1, 2, 3, 4, 5, 9, 10, 11, 12, 14, 15 point somewhat toward the lidar sensor. And thus, the frontal features of the car appear: Windshield, doors, roof, hood, bumper.
The yaw angles of cars 6, 7, 8 show the lateral features of them: Doors, roof, hood, bumper.
Only the car 13 is heading in the opposite direction of the lidar sensor, showing the back features: Rear bumper, trunk, roof.
python3 loop_over_dataset_ID_S2_EX1.py
python3 loop_over_dataset_ID_S2_EX2.py
python3 loop_over_dataset_ID_S2_EX3.py
python3 loop_over_dataset_ID_S3_EX1.py
python3 loop_over_dataset_ID_S3_EX2.py
python3 loop_over_dataset_ID_S4_EX1.py
ious=[0.6542068636263976, 0.9132875663091978, 0.8925967031816004]
center_devs=[[0.311355588504739, -0.0031356753934232984, 1.0292643213596193], [-0.07108406474071671, 0.03811673568270635, 0.8291298942401681], [0.07775295163446572, 0.030665009917356656, 0.8929607095304846]]
ious=[0.7096651012386277, 0.8999471797245473, 0.896371116728743]
center_devs=[[0.2488309520376788, -0.029714277508901432, 0.9899635729132115], [-0.08442348191056226, 0.04014699931940413, 0.8152483974641882], [-0.0872158711317752, -0.02108056463293906, 0.8711946193956237]]
python3 loop_over_dataset_ID_S4_EX2.py
python3 loop_over_dataset_ID_S4_EX3_darknet.py
student task ID_S4_EX3
all_positives=306, true_positives=287, false_negatives=19, false_positives=14
precision = 0.9534883720930233, recall = 0.9379084967320261
python3 loop_over_dataset_ID_S4_EX3_fpn_resnet.py
student task ID_S4_EX3
all_positives=306, true_positives=139, false_negatives=167, false_positives=103
precision = 0.5743801652892562, recall = 0.4542483660130719
python3 loop_over_dataset_ID_S4_EX3_labels.py
student task ID_S4_EX3
all_positives=306, true_positives=306, false_negatives=0, false_positives=0
precision = 1.0, recall = 1.0
This is the project for the second course in the Udacity Self-Driving Car Engineer Nanodegree Program : Sensor Fusion and Tracking.
In this project, you'll fuse measurements from LiDAR and camera and track vehicles over time. You will be using real-world data from the Waymo Open Dataset, detect objects in 3D point clouds and apply an extended Kalman filter for sensor fusion and tracking.
The project consists of two major parts:
- Object detection: In this part, a deep-learning approach is used to detect vehicles in LiDAR data based on a birds-eye view perspective of the 3D point-cloud. Also, a series of performance measures is used to evaluate the performance of the detection approach.
- Object tracking : In this part, an extended Kalman filter is used to track vehicles over time, based on the lidar detections fused with camera detections. Data association and track management are implemented as well.
The following diagram contains an outline of the data flow and of the individual steps that make up the algorithm.
Also, the project code contains various tasks, which are detailed step-by-step in the code. More information on the algorithm and on the tasks can be found in the Udacity classroom.
📦project
┣ 📂dataset --> contains the Waymo Open Dataset sequences
┃
┣ 📂misc
┃ ┣ evaluation.py --> plot functions for tracking visualization and RMSE calculation
┃ ┣ helpers.py --> misc. helper functions, e.g. for loading / saving binary files
┃ ┗ objdet_tools.py --> object detection functions without student tasks
┃ ┗ params.py --> parameter file for the tracking part
┃
┣ 📂results --> binary files with pre-computed intermediate results
┃
┣ 📂student
┃ ┣ association.py --> data association logic for assigning measurements to tracks incl. student tasks
┃ ┣ filter.py --> extended Kalman filter implementation incl. student tasks
┃ ┣ measurements.py --> sensor and measurement classes for camera and lidar incl. student tasks
┃ ┣ objdet_detect.py --> model-based object detection incl. student tasks
┃ ┣ objdet_eval.py --> performance assessment for object detection incl. student tasks
┃ ┣ objdet_pcl.py --> point-cloud functions, e.g. for birds-eye view incl. student tasks
┃ ┗ trackmanagement.py --> track and track management classes incl. student tasks
┃
┣ 📂tools --> external tools
┃ ┣ 📂objdet_models --> models for object detection
┃ ┃ ┃
┃ ┃ ┣ 📂darknet
┃ ┃ ┃ ┣ 📂config
┃ ┃ ┃ ┣ 📂models --> darknet / yolo model class and tools
┃ ┃ ┃ ┣ 📂pretrained --> copy pre-trained model file here
┃ ┃ ┃ ┃ ┗ complex_yolov4_mse_loss.pth
┃ ┃ ┃ ┣ 📂utils --> various helper functions
┃ ┃ ┃
┃ ┃ ┗ 📂resnet
┃ ┃ ┃ ┣ 📂models --> fpn_resnet model class and tools
┃ ┃ ┃ ┣ 📂pretrained --> copy pre-trained model file here
┃ ┃ ┃ ┃ ┗ fpn_resnet_18_epoch_300.pth
┃ ┃ ┃ ┣ 📂utils --> various helper functions
┃ ┃ ┃
┃ ┗ 📂waymo_reader --> functions for light-weight loading of Waymo sequences
┃
┣ basic_loop.py
┣ loop_over_dataset.py
In order to create a local copy of the project, please click on "Code" and then "Download ZIP". Alternatively, you may of-course use GitHub Desktop or Git Bash for this purpose.
The project has been written using Python 3.7. Please make sure that your local installation is equal or above this version.
All dependencies required for the project have been listed in the file requirements.txt
. You may either install them one-by-one using pip or you can use the following command to install them all at once:
pip3 install -r requirements.txt
The Waymo Open Dataset Reader is a very convenient toolbox that allows you to access sequences from the Waymo Open Dataset without the need of installing all of the heavy-weight dependencies that come along with the official toolbox. The installation instructions can be found in tools/waymo_reader/README.md
.
This project makes use of three different sequences to illustrate the concepts of object detection and tracking. These are:
- Sequence 1 :
training_segment-1005081002024129653_5313_150_5333_150_with_camera_labels.tfrecord
- Sequence 2 :
training_segment-10072231702153043603_5725_000_5745_000_with_camera_labels.tfrecord
- Sequence 3 :
training_segment-10963653239323173269_1924_000_1944_000_with_camera_labels.tfrecord
To download these files, you will have to register with Waymo Open Dataset first: Open Dataset – Waymo, if you have not already, making sure to note "Udacity" as your institution.
Once you have done so, please click here to access the Google Cloud Container that holds all the sequences. Once you have been cleared for access by Waymo (which might take up to 48 hours), you can download the individual sequences.
The sequences listed above can be found in the folder "training". Please download them and put the tfrecord
-files into the dataset
folder of this project.
The object detection methods used in this project use pre-trained models which have been provided by the original authors. They can be downloaded here (darknet) and here (fpn_resnet). Once downloaded, please copy the model files into the paths /tools/objdet_models/darknet/pretrained
and /tools/objdet_models/fpn_resnet/pretrained
respectively.
In the main file loop_over_dataset.py
, you can choose which steps of the algorithm should be executed. If you want to call a specific function, you simply need to add the corresponding string literal to one of the following lists:
-
exec_data
: controls the execution of steps related to sensor data.pcl_from_rangeimage
transforms the Waymo Open Data range image into a 3D point-cloudload_image
returns the image of the front camera
-
exec_detection
: controls which steps of model-based 3D object detection are performedbev_from_pcl
transforms the point-cloud into a fixed-size birds-eye view perspectivedetect_objects
executes the actual detection and returns a set of objects (only vehicles)validate_object_labels
decides which ground-truth labels should be considered (e.g. based on difficulty or visibility)measure_detection_performance
contains methods to evaluate detection performance for a single frame
In case you do not include a specific step into the list, pre-computed binary files will be loaded instead. This enables you to run the algorithm and look at the results even without having implemented anything yet. The pre-computed results for the mid-term project need to be loaded using this link. Please use the folder darknet
first. Unzip the file within and put its content into the folder results
.
-
exec_tracking
: controls the execution of the object tracking algorithm -
exec_visualization
: controls the visualization of resultsshow_range_image
displays two LiDAR range image channels (range and intensity)show_labels_in_image
projects ground-truth boxes into the front camera imageshow_objects_and_labels_in_bev
projects detected objects and label boxes into the birds-eye viewshow_objects_in_bev_labels_in_camera
displays a stacked view with labels inside the camera image on top and the birds-eye view with detected objects on the bottomshow_tracks
displays the tracking resultsshow_detection_performance
displays the performance evaluation based on all detectedmake_tracking_movie
renders an output movie of the object tracking results
Even without solving any of the tasks, the project code can be executed.
The final project uses pre-computed lidar detections in order for all students to have the same input data. If you use the workspace, the data is prepared there already. Otherwise, download the pre-computed lidar detections (~1 GB), unzip them and put them in the folder results
.
Parts of this project are based on the following repositories: