search

zylo117 / Yet-Another-EfficientDet-Pytorch

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.
GNU Lesser General Public License v3.0
5.21k stars 1.27k forks source link

efficientdet_test.py for custom dataset #698

Open Kislunoman opened 2 years ago

Kislunoman commented 2 years ago

I have trained my model on a custom dataset. Please let me know if anybody applies the efficientdet_test.py for a trained model on test images. How do you edit this file ?

zylo117 commented 2 years ago
  1. init model
  2. load the weights
  3. perprocess, infer and postprocess the image
  4. draw boxes on images