Skip to content

Latest commit

 

History

History
31 lines (27 loc) · 1008 Bytes

README.md

File metadata and controls

31 lines (27 loc) · 1008 Bytes
Raw

Pedestrian_Detection

Description

An occluded pedestrian detection project using improved YOLOv5s model

Key Features

  • Improved YOLOv5s model to enhance occluded object detection capabilities by embedding SE attention mechanisms and replacing SPP with SPP-Fast modules in backbone and adding specialized detection head for small objects
  • Constructed and trained improved model in PyTorch based on dataset of 2,000+ images with occluded pedestrians

Tech Stack

  • Data Processing: Python - Pandas, Numpy
  • Modeling: Python - PyTorch

Project Structure

Pedestrian Detection
├── data/    # trajectory dataset
├── models/
├── runs/
├── utils/
├── venv/
├── src/
|   ├── detect.py   # detection
|   ├── export.py   # export model to other formats
|   ├── train.py    # train
|   └── val.py      # validate
├── requirements.txt
├── README.md
└── LICENSE

Feel free to open issues and pull requests to contribute!