__init__.py

from . import rpn_head
from . import bbox_head
from . import mask_head
from . import yolo_head

from .rpn_head import *
from .bbox_head import *
from .mask_head import *
from .yolo_head import *