NGC | Catalog
Welcome Guest
CatalogModelsEfficientDet-D0 checkpoint (Pytorch, AMP, COCO17)

EfficientDet-D0 checkpoint (Pytorch, AMP, COCO17)

For downloads and more information, please view on a desktop device.
Logo for EfficientDet-D0 checkpoint (Pytorch, AMP, COCO17)

Description

EfficientDet-D0 Pytorch checkpoint trained on COCO using batchsize=1200

Publisher

NVIDIA Deep Learning Examples

Use Case

Detection

Framework

Pytorch

Latest Version

21.06.0

Modified

October 29, 2021

Size

46.97 MB

Model Overview

A convolution-based neural network for the task of object detection

Model Architecture

EfficientDet is a one-stage detector with the following architecture components:

  • ImageNet-pretrained EfficientNet backbone
  • Weighted bi-directional feature pyramid network (BiFPN)
  • Bounding and classification box head
  • A compound scaling method that uniformly scales the resolution, depth, and width for all backbone, feature network, and box/class prediction networks at the same time

Training

This model was trained using script available on NGC and in GitHub repo

Dataset

The following datasets were used to train this model:

  • COCO 2017 - Dataset for large-scale object detection, segmentation and captioning.

Performance

Performance numbers for this model are available in NGC

References

License

This model was trained using open-source software available in Deep Learning Examples repository. For terms of use, please refer to the license of the script and the datasets the model was derived from.