https://github.com/kentaroy47/efficientdet.pytorch
:neckbeard:Unofficial implementation of EfficientDet
https://github.com/kentaroy47/efficientdet.pytorch
apex efficientdet pytorch retinanet ssd
Last synced: 8 months ago
JSON representation
:neckbeard:Unofficial implementation of EfficientDet
- Host: GitHub
- URL: https://github.com/kentaroy47/efficientdet.pytorch
- Owner: kentaroy47
- License: mit
- Created: 2019-11-25T00:57:28.000Z (almost 6 years ago)
- Default Branch: master
- Last Pushed: 2020-03-11T08:30:50.000Z (over 5 years ago)
- Last Synced: 2025-01-20T23:52:40.676Z (9 months ago)
- Topics: apex, efficientdet, pytorch, retinanet, ssd
- Language: Jupyter Notebook
- Homepage:
- Size: 44.4 MB
- Stars: 63
- Watchers: 8
- Forks: 16
- Open Issues: 7
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
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 | | |
# efficientdet.pytorch
unofficial implementation of efficientdet
- supported efficientnet backbone (B0 for now)
- supported resnet backbone.
- supports BiFPN.
- supports VOC and COCO training.
## requirements
`pip install --upgrade efficientnet-pytorch`
## PASCAL-VOC training
- see `train_efficientdet.ipynb`
- see `eval.ipynb` for evaluation
## COCO training
- see `train_efficientdet.ipynb` and configure MS-COCO dataset.
## FP16 training
- see `train_fp16_apex.ipynb`
