Skip to content

sonalrpatel/parking-slot-detection

Repository files navigation

parking-slot-detection

Introduction

References

Folder Structure

Sample Results

TODOs

Introduction

This repository is an implementation of parking slot detection in AVM (around view images) using deep learning. The implementation is based on following References.

References

Paper

around view image

parking lot image

Github

around view image

object detection using yolo

Blog

around view image

parking lot image

Dataset

around view image

Folder Structure

Explanation about folders and files.

  • data - contains the datasets, annotation files, and class details
    • {dataset folder name}
      • train
      • val
      • test
      • train_annotation.txt
      • val_annotation.txt
      • test_annotation.txt
    • ps_classes.txt
  • dataloader
    • dataloader.py - custom data generator
  • loss
    • loss_functional.py - loss is written in a function
    • loss_subclass.py - loss is written under a class
  • model
    • darknet.py - backbone
    • model_functional.py - functional model
    • model_subclass.py - model sub-classing
  • model_yolo3_tf2 - yolov3 model reference from yolov3-tf2
  • model_data - contains the .cfg, .weights, .h5, font files
  • model_img - contains the architecture images
  • notebook - contains the jupyter / google colab notebook file
  • utils
    • callbacks.py
    • dataloader.py - reference from yolov3-tf2
    • utils.py
    • utils_bbox.py
    • utils_metric.py
  • configs.py
  • convert.py
  • predict.py
  • train.py
  • convert0.py - reference from yolov3-tf2
  • predict0.py - reference from yolov3-tf2
  • train0.py - reference from yolov3-tf2
  • psmat2txt.py - generates annotations in YOLO v3 Keras TXT format from mat files
  • visbbox.py - visualise bounding boxes

Sample Results

Sample results of parking slot head detection with VPS-Net as reference

Parking slot head Parking slot head

TODOs

  • find complete parking slot
  • improve performance by referring other papers