__init__.py 696 B

12345678910111213141516
  1. from .bifpn import BiFPN
  2. from .efficientdet import EfficientDet
  3. from .efficientdet_head import EfficientDetSepBNHead
  4. from .huber_loss import HuberLoss
  5. from .tensorflow.anchor_generator import YXYXAnchorGenerator
  6. from .tensorflow.coco_90class import Coco90Dataset
  7. from .tensorflow.coco_90metric import Coco90Metric
  8. from .tensorflow.trans_max_iou_assigner import TransMaxIoUAssigner
  9. from .tensorflow.yxyx_bbox_coder import YXYXDeltaXYWHBBoxCoder
  10. from .utils import Conv2dSamePadding
  11. __all__ = [
  12. 'EfficientDet', 'BiFPN', 'HuberLoss', 'EfficientDetSepBNHead',
  13. 'Conv2dSamePadding', 'Coco90Dataset', 'Coco90Metric',
  14. 'YXYXAnchorGenerator', 'TransMaxIoUAssigner', 'YXYXDeltaXYWHBBoxCoder'
  15. ]