:neckbeard:Unofficial implementation of EfficientDet
Now training models..
backbone | resolution | VOCmAP | COCOmAP | Inference[ms] | model |
---|---|---|---|---|---|
EfficientnetB0(wo/BiFPN) | 512 | 77.0 | TBD | ||
EfficientnetB0(w/BiFPN) | 512 | 77.2 | TBD | ||
EfficientnetB2(wo/BiFPN) | 768 | TBD | TBD | ||
EfficientnetB2(w/BiFPN) | 768 | TBD | TBD | ||
EfficientnetB4(wo/BiFPN) | 1024 | TBD | TBD | ||
EfficientnetB4(w/BiFPN) | 1024 | TBD | TBD | ||
EfficientnetB6(wo/BiFPN) | 1408 | TBD | TBD | ||
EfficientnetB6(w/BiFPN) | 1408 | TBD | TBD |
unofficial implementation of efficientdet
pip install --upgrade efficientnet-pytorch
train_efficientdet.ipynb
eval.ipynb
for evaluationtrain_efficientdet.ipynb
and configure MS-COCO dataset.train_fp16_apex.ipynb