# Implementation in this file is modified based on ViTAE-Transformer
# Originally Apache 2.0 License and publicly available at https://github.com/ViTAE-Transformer/ViTDet
from .anchor_head import AnchorNHead
from .rpn_head import RPNNHead

__all__ = ['AnchorNHead', 'RPNNHead']
