This repository contains pretrained EfficientDet model which detects character heads in illustrations. Models are trained using zylo117's EfficientDet implementation and can be tested using the implementation.
Models are trained on a dataset with manual annotations by myself.
#Train Images = 3184
#Val Images = 197
| coefficient | Download URL | mAP (IoU:0.5-0.95) | recall (IoU:0.5-0.95) | Note |
|---|---|---|---|---|
| D0 | character-face-efficientdet-c0.pth | 0.713 | 0.781 | Tested with PyTorch==1.5.0 |
| D2 | character-face-efficientdet-c2.pth | 0.767 | 0.812 | Tested with Pytorch==1.5.0 |