Skip to content

PyTorch implementation of ComplexYOLO. This Repository is meant for students to learn object detection and working with LiDAR/RADAR datasets.

License

Notifications You must be signed in to change notification settings

BerensRWU/Complex_YOLO

Repository files navigation

ComplexYOLO

This repository contains a PyTorch implementation of ComplexYOLO using YOLO version 3. It is build to be applied on the data from the Astyx Dataset.

Installation

Requirements

The repo was tested on the following systems with the following versions:

Ubuntu 18.04.5 LTS

To run the program you need to install those libraries with dependencies:

  • torch (Tested with pytorch version 1.6.0, torchvision 0.7.0)
  • cv2 (Tested with opencv-python 4.1.1)
  • json
  • numpy
  • shapely

Windows 10

To run the program you need to install those libraries with dependencies:

  • torch (Tested with pytorch version 1.5.1, torchvision 0.6.1)
  • cv2
  • json
  • numpy
  • shapely

Google Colab

In google colab you do not need to install any libraries. But you will need a google account. If you want to use colab you click on "main.ipynb" and then on "open in colab". After you loged into your google account, you have to upload the scripts and weights to the files in the notebook. Because of the size the Astyx data should be uploaded to your drive and change in utils->config->root_dir the root directory to drive/My Drive/dataset or wherever you saved the dataset.

Instead of flags we can specify in the fourth cell what we want to do. visualize=True then we will visualize the data, estimate_bb=True the bounding boxes will be estimated, evaluate=True the performance of the detector will be evaluated (Then estimate_bb must also be true). For RADAR data we have to set radar=True.

Installation for working on your machine

An easy way to install libaries on Windows and Linux is with anaconda or miniconda. Then you can install e.g. torch:

conda install pytorch torchvision cpuonly -c pytorch

Steps

  1. Install all requirements
  2. Download or clone this repo by using git clone https://github.com/BerensRWU/ComplexYOLO/ in the terminal.
  3. Save the Astyx dataset in the folder dataset.(See Section Astyx HiRes).
  4. Run main.py --visualize to visualize the ground truth data of the validation split from the Astyx dataset.
  5. Download the weights for the RADAR and LiDAR detector from the moodle page of the Lecture.
  6. Save the weights files in a folder checkpoints in the Complex_YOLO folder.
  7. Write the iou function in utils.py.
  8. Run main.py --estimate_bb to estimate the bounding boxes.
  9. Run main.py --estimate_bb --visualize to estimate the bounding boxes and to visualize the estimation together with the ground truth.
  10. Run main.py --estimate_bb --evaluate to estimate the bounding boxes and to evaluate them.
  11. If we want to use RADAR data and detector instead of LiDAR we have to use the flag radar. E.g. main.py --estimate_bb --evaluate --radar will evaluate the performance of the RADAR detector.

Scripts

main.py.

python main.py

Arguments:

  • estimate_bb: Flag, to estimate bounding boxes. Default False.
  • radar: Flag, if True use radar data otherwise lidar. Default False.
  • model_def: Path to model definition file. Default config/yolov3-custom.cfg.
  • weights_path: Path where the weights are saved. Default checkpoints.
  • conf_thres: If bounding boxes are estimated, only those with a confidence greater than thethreshold will be used. Default 0.5.
  • iou_thres: If bounding boxes are estimated, threshold for a correct detection. Default 0.5.
  • nms_thres: If estimated bounding boxes overlap with an IoU greater than the nms_thres only the bounding box with highest confidence remains. Default 0.2.
  • split: Which split to use valid, train. Default valid.
  • visualize: Whether to visualize the data.
  • evaluate: Whether to evaluate the data.

visualize.py

A script for the visualization of the data.

detector.py

A script for the detection of bounding boxes from BEV.

evaluation.py

A script for the evaluation of the predicted bounding boxes. Calculation whether the prediction is a true positive or not and the average precision.

How to use this Repo for the LiDAR RADAR lecture

If you want to use this repository for the LiDAR RADAR lecture, you have two ways:

Terminal

To run on your own machine in the terminal, download this repository. Install python and all requirements. Define the path to the data in utils/config.py it should have the order like in section Astyx HiRes. Now we can run the main function in the terminal:

python main.py

This will check if you have defined the correct path to the data.

In the next step we will visualize the data:

python main.py --visualize

This will save the BEVs and the ground truth data in the folder output. In default this will use LiDAR data, if we want to use radar data we have to add the flag --radar. The coordinates of the ground truth are in the RADAR coordinatesystem. Because of this the LiDAR data is transformed to the RADAR coordinatesystem.

Next we want to predict bounding boxes. For this we need to download the weights for the network and store them in the folder checkpoints, one for RADAR and one for LiDAR. Also we need non maximum supression, such that predicted bounding boxes that overlap get merged to one. For the non maximum supression we need the intersection over union, to write this function is your task. You find a starting point in utils/utils.py in the function def compute_iou(box, boxes) it gets one bounding box as shapley polygon and a list of shapley polygones and returns a list of the corresponding iou values. Then the prediction can be done with the following command:

python main.py --estimate_bb

To visualize and predict we can use the following command:

python main.py --estimate_bb --visualize

To evaluate the predictions we use the average precision, so we need next to a function for the IoU a function for the average precision. In the script evaluation.py you find funcitons for this.

To evaluate the prediction we can use the following command:

python main.py --estimate_bb --evaluate

Astyx HiRes

The Astyx HiRes is a dataset from Astyx for object detection for autonomous driving. Astyx has a sensor setup consisting of camera, LiDAR, RADAR. Additional information can be found here: Dataset Paper and Specification

└── dataset/
       ├── dataset_astyx_hires2019    <-- 546 data
       |   ├── calibration 
       |   ├── camera_front
       |   ├── groundtruth_obj3d
       |   ├── lidar_vlp16
       |   ├── radar_6455
       └── split
           ├── train.txt
           └── valid.txt

About

PyTorch implementation of ComplexYOLO. This Repository is meant for students to learn object detection and working with LiDAR/RADAR datasets.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published