/EfficientDet

Repo for testing out effdet pytorch version

Primary LanguageJupyter Notebook

Effcient Det Pytorch

This repo was used as a test to build effcientnetV2 for object detection, using the fruits object detection dataset as a base.

This repo will include from data exploration all the way to deployment.

Installation

To run training code

python train.py

ToDO

  • EDA
  • Train
  • Evaluation using sklearn
  • Inference Script
  • Deploy to triton and fastapi

ToDO - Cleanup

  • Convert notebook to py scripts for easier viewing
  • Update training with metrics

References