Exemplo n.º 1
0
def main():
    parser = create_argument_parser()
    args = parser.parse_args()
    verbosity = args.verbosity if hasattr(args, "verbosity") else None
    global logger
    logger = setup_logger(name=LOGGER_NAME)
    logger.setLevel(verbosity_to_level(verbosity))
    args.func(args)
Exemplo n.º 2
0
def main():
    parser = create_argument_parser()
    parser.add_argument("--adaptive-pool",
                        action="store_true",
                        help="replace roialign, roipool to adaptivepool")
    args = parser.parse_args()

    if args.adaptive_pool:
        import detectron2.modeling.poolers
        detectron2.modeling.poolers.RoIPool = AdaptivePool
        detectron2.modeling.poolers.ROIAlign = AdaptivePool

    verbosity = args.verbosity if hasattr(args, "verbosity") else None
    global logger
    logger = setup_logger(name=LOGGER_NAME)
    logger.setLevel(verbosity_to_level(verbosity))
    args.func(args)