Here is our pytorch implementation of the EfficientDet model to Detection and classification of small objects
Our project running on Colab pro
Here is our pytorch implementation of the model described in the Paper
EfficientDet: Scalable and Efficient Object Detection paper (Note: We also provide pre-trained weights, which you could see in Drive at folder "train_weights"
What You Will Learn
How to load your custom image detection data from Roboflow
How to instatiate a pytorch EfficientDet model
How to train the EfficientDet model
How to use the model for quick inference
How to export the model weights for future inference
We trained our model by using Colab Pro . Below is mAP (mean average precision) for Chess and All UAVs (MAV-VID,Full-UAV and Combine of them) datasets under IoU=0.50