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)
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)