INNER CODE UNIT · Python
build_anchor_generator
apple/ml-cvnets · cvnets/anchor_generator/__init__.py:33
def build_anchor_generator(opts, *args, **kwargs):
"""Build anchor generator for object detection"""
anchor_gen_name = getattr(opts, "anchor_generator.name")
# We registered the base class using a special `name` (i.e., `__base__`)
# in order to access the arguments defined inside those classes. However, these classes are not supposed to
# be used. Therefore, we raise an error for such cases
if anchor_gen_name == "__base__":
logger.error("__base__ can't be used as a projection name. Please check.")
anchor_gen = ANCHOR_GEN_REGISTRY[anchor_gen_name](opts, *args, **kwargs)
return anchor_gen