add symlink method

mmcv/__init__.py

@@ -5,4 +5,4 @@ from .opencv_info import *
 from .image import *
 from .video import *
 from .visualization import *
-from .version import *
+from .version import __version__

mmcv/utils/path.py

@@ -33,6 +33,9 @@ def check_file_exist(filename, msg_tmpl='file "{}" does not exist'):
 
 
 def mkdir_or_exist(dir_name, mode=0o777):
+    if dir_name == '':
+        return
+    dir_name = osp.expanduser(dir_name)
     if six.PY3:
         os.makedirs(dir_name, mode=mode, exist_ok=True)
     else:
@@ -40,6 +43,12 @@ def mkdir_or_exist(dir_name, mode=0o777):
             os.makedirs(dir_name, mode=0o777)
 
 
+def symlink(src, dst, overwrite=True, **kwargs):
+    if os.path.lexists(dst) and overwrite:
+        os.remove(dst)
+    os.symlink(src, dst, **kwargs)
+
+
 def _scandir_py35(dir_path, suffix=None):
     for entry in os.scandir(dir_path):
         if not entry.is_file():