diff --git a/mmdet/utils/__init__.py b/mmdet/utils/__init__.py index 5a384feafdf..b5a2b6b3a0e 100644 --- a/mmdet/utils/__init__.py +++ b/mmdet/utils/__init__.py @@ -7,6 +7,7 @@ from .memory import AvoidCUDAOOM, AvoidOOM from .misc import find_latest_checkpoint, update_data_root from .replace_cfg_vals import replace_cfg_vals +from .rfnext import rfnext_init_model from .setup_env import setup_multi_processes from .split_batch import split_batch from .util_distribution import build_ddp, build_dp, get_device @@ -16,5 +17,6 @@ 'update_data_root', 'setup_multi_processes', 'get_caller_name', 'log_img_scale', 'compat_cfg', 'split_batch', 'build_ddp', 'build_dp', 'get_device', 'replace_cfg_vals', 'AvoidOOM', 'AvoidCUDAOOM', - 'get_max_num_gt_division_factor', 'masked_fill', 'batch_images_to_levels' + 'get_max_num_gt_division_factor', 'masked_fill', 'batch_images_to_levels', + 'rfnext_init_model' ]