← all repositories

rwightman/efficientdet-pytorch

A PyTorch implementation of Google's EfficientDet object detection model with configurable BiFPN architecture and training pipelines.

1.7k stars Python Computer VisionML Frameworks
efficientdet-pytorch
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository provides a faithful PyTorch port of the original TensorFlow EfficientDet architecture for scalable object detection. It implements BiFPN (Bidirectional Feature Pyramid Network) connections with fully configurable combination modes, supports any backbone from the timm model collection that supports feature extraction, and includes training and validation scripts with PyTorch 2.0 torchcompile support. The implementation allows switching between depthwise separable and standard convolutions for the BiFPN and detection head modules.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.