A vacant parking slot detection method in the around view image based on deep learning.
We ran our experiments with PyTorch 1.0.1, CUDA 9.0, Conda with Python 3.6 and Ubuntu 16.04.
$ git clone https://github.com/weili1457355863/VPS-Net.git
$ cd VPS-Net
$ conda create --name vps-net python=3.6
$ conda activate vps-net
$ pip install -r requirements.txt
$ mkdir weights
$ cd weights/
Download the weights of detection network and classification network.
$ mkdir data
$ cd data/
Download the ps2.0 dataset or the PSV dataset
Uses pretrained weights to detect the vacant parking slot in the around view image.
$ vps_net.py [-h] [--input_folder INPUT_FOLDER]
[--output_folder OUTPUT_FOLDER] [--model_def MODEL_DEF]
[--weights_path_yolo WEIGHTS_PATH_YOLO]
[--weights_path_vps WEIGHTS_PATH_VPS]
[--conf_thres CONF_THRES] [--nms_thres NMS_THRES]
[--img_size IMG_SIZE] [--save_files SAVE_FILES]
Test on the ps2.0 dataset. The detection results including images and files will be saved.
$ python vps_net.py --input_folder data/ps2.0/testing/all --save_files 1
In order to facilitate other researchers, the annoation for vacant parking slots of ps 2.0 and PSV datasets has been made publicly avaliable.
@article{li_vacant_2020,
title = {Vacant Parking Slot Detection in the Around View Image Based on Deep Learning},
volume = {20},
doi = {10.3390/s20072138},
pages = {2138},
journal = {Sensors},
author = {Li, Wei and Cao, Libo and Yan, Lingbo and Li, Chaohui and Feng, Xiexing and Zhao, Peijie},
date = {2020-04-10}
}