efficientdet.pytorch

unofficial implementation of efficientdet

requirements

pip install --upgrade efficientnet-pytorch

please see train_efficientdet.ipynb for VOC training.

Now training models.. Plz wait.

backbone	resolution	VOCmAP	COCOmAP
EfficientnetB0(wo/BiFPN)	512	77.0	TBD
EfficientnetB0(w/BiFPN)	512	79.0	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

Name		Name	Last commit message	Last commit date
Latest commit History 31 Commits
VOC2007		VOC2007
dataset		dataset
utils		utils
weights		weights
BiFPN.py		BiFPN.py
README.md		README.md
effnet_testing.ipynb		effnet_testing.ipynb
effnet_testing.py		effnet_testing.py
eval.ipynb		eval.ipynb
eval.py		eval.py
fire-video.mp4		fire-video.mp4
log_output.csv		log_output.csv
play_vedio.py		play_vedio.py
testnet.py		testnet.py
testnet.sh		testnet.sh
testnet_vedio.py		testnet_vedio.py
testnet_vedio.sh		testnet_vedio.sh
train_apex.ipynb		train_apex.ipynb
train_apex.py		train_apex.py
train_efficientdet-ooo.py		train_efficientdet-ooo.py
train_efficientdet.ipynb		train_efficientdet.ipynb
train_efficientdet.py		train_efficientdet.py
train_fire.py		train_fire.py
train_fire.sh		train_fire.sh
train_retinanet.py		train_retinanet.py