rwightman/efficientdet-pytorch
A PyTorch implementation of Google's EfficientDet object detection model with configurable BiFPN architecture and training pipelines.

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.