minz27 / aerial-pedestrian-detection-pytorch

FlyAI

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

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