Train YOLOv2 with KITTI dataset

Yizhou Wang, Zhichao Lei   July 29, 2018   0 Comments


GitHub repository: https://github.com/yizhou-wang/darknet-kitti.

KITTI dataset contains many real-world computer vision benchmarks for autonomous driving. There are many tasks including stereo, optical flow, visual odometry, 3D object detection and 3D tracking. YOLOv2 is a popular technique for real-time object detection. There are many pre-trained weights for many current image datasets. However, YOLOv2 doesn’t perform well on KITTI object dataset. In this post, I will explain how to train YOLOv2 with KITTI object dataset and show some test results using our trained weights.


Prepare KITTI dataset

We used KITTI object 2D for training YOLO and used KITTI raw data for test. Some of the test results are recorded as the demo video above.

Download data and labels

Download KITTI object 2D left color images of object data set (12 GB) and submit your email address to get the download link. Download training labels of object data set (5 MB). Unzip them to your customized directory <data_dir> and <label_dir>.

Convert KITTI labels to YOLO labels

To simplify the labels, we combined 9 original KITTI labels into 6 classes:

Car
Van
Truck
Tram
Pedestrian
Cyclist

Need to refer the script from Zhichao.

Why is KITTI difficult to train on YOLO?

Many people tried to train YOLOv2 with KITTI dataset but often get really poor performance. This is a typical result of YOLOv2 detection without doing any modification. This is a YOLOv2 trained on 3 classes of KITTI dataset.

Why does YOLOv2 perform bad on KITTI unlike other datasets? After review the basic properties of KITTI, we can find that the shape of the images is really wide: \(1224 \times 370\). However, the default input shape of YOLOv2 is \(416 \times 416\). After this kind of resizing, the bbox of the object would because really thin, and probably result in the bad performance. Moreover, the sizes of the objects in KITTI could be various. Some of the objects could be too small to be detected.

Configuration settings

There are two ways of configuration:

  1. Change the input shape of YOLOv2 model and disable random resizing.
  2. Modify the resizing code in YOLOv2 source code.

Change the input shape

Open the configuration file yolov2-voc.cfg and change the following parameters:

[net]
# Training
batch=64
subdivisions=8
height=370
width=1224

[region]
classes=6

random=0

Also, remember to change the filters in the last convolutional layer to be \(\texttt{filters} = ((\texttt{classes} + 5) \times \texttt{num})\), so that

# last convolutional layer
[convolutional]
filters=55

You can also refine some other parameters like learning_rate, object_scale, thresh, etc. to obtain even better results.

Our configuration file kitti6-yolov2.cfg for KITTI with 6 classes can be found HERE.

Modify the resizing code

TODO

Training results

The training results statistics are shown in the table below.

Object Average Recall mAP
Overall
Car 96.8%
Van 95.8%
Truck 99.4%
Tram 99.0%
Pedestrian 86.1%
Cyclist 91.3%

Test on KITTI image sequences

We wrote a new function in darknet named testseq, which can test YOLO performance for a image sequence. The file names of the image sequence should be listed in a txt file <namelist.txt>. To use testseq, run the following code in the terminal:

./darknet detector testseq cfg/kitti.data cfg/kitti.cfg <weights_file> <namelist.txt> 

The output results should look like the demo video on the top.

YOLOv3 could perform better on KITTI and we will post the training process and results in one of the following posts.


deep-learning  object-detection  kitti  yolo