Skip to content

Training SSD: Single Shot MultiBox Detector with FastAI library

Notifications You must be signed in to change notification settings

lanhhv84/FastAI-SSD

Repository files navigation

Introduction

Result

Todo

  • Read Pascal VOC with FastAI DataBunch
  • Create modified VGG architecture with pure Pytorch and Python code
  • Write loss function with pure Pytorch and Python code
  • Train with FastAI api
  • Inferent pictures
  • Non-Max Suppression
  • mAP (mean Average Precision)
  • Inferent with video
  • Real time inference
  • Clean code (Optional)

Problems

  • Slow IOU calculation
  • Slow inference

About

Training SSD: Single Shot MultiBox Detector with FastAI library

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published