Skip to content

minz27/aerial-pedestrian-detection-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FlyAI

Fine-tuning a pre-trained object detection model for the Standford Drone Dataset.

img

Code in sdd-utils comes from here.

Tensorflow-Lite Approach

Requirements & Dataset extraction

  • Requires ffmpeg to be installed.
pip install -r requirements.txt

and for to create images from video frames and create Pascal VOC dataset.

sdd-utils/annotate.py

PyTorch Approach

cd pytorch

Requirements

pip install -r requirements.txt

Run

Generate images from videos

python src/utils_.py

Train

python src/main.py

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages