diff --git a/DOTA_devkit/polyiou.py b/DOTA_devkit/polyiou.py
index 00ff746..10f18e9 100644
--- a/DOTA_devkit/polyiou.py
+++ b/DOTA_devkit/polyiou.py
@@ -12,16 +12,16 @@
 if version_info >= (2, 6, 0):
     def swig_import_helper():
         from os.path import dirname
-        import imp
+        import importlib
         fp = None
         try:
-            fp, pathname, description = imp.find_module('_polyiou', [dirname(__file__)])
+            fp, pathname, description = importlib.find_module('_polyiou', [dirname(__file__)])
         except ImportError:
             import _polyiou
             return _polyiou
         if fp is not None:
             try:
-                _mod = imp.load_module('_polyiou', fp, pathname, description)
+                _mod = importlib.load_module('_polyiou.cpython-37m-x86_64-linux-gnu.so', fp, pathname, description)
             finally:
                 fp.close()
             return _mod
diff --git a/DOTA_devkit/prepare_dota1.py b/DOTA_devkit/prepare_dota1.py
index bc55723..11c2b78 100644
--- a/DOTA_devkit/prepare_dota1.py
+++ b/DOTA_devkit/prepare_dota1.py
@@ -12,8 +12,8 @@
 
 def parse_args():
     parser = argparse.ArgumentParser(description='prepare dota1')
-    parser.add_argument('--srcpath', default='/home/dingjian/project/dota')
-    parser.add_argument('--dstpath', default=r'/home/dingjian/workfs/dota1-split-1024',
+    parser.add_argument('--srcpath', default='/content/AerialDetection/data/dota')
+    parser.add_argument('--dstpath', default=r'/content/AerialDetection/data/dota1-split-1024',
                         help='prepare data')
     args = parser.parse_args()
 
@@ -99,4 +99,4 @@ def prepare(srcpath, dstpath):
     args = parse_args()
     srcpath = args.srcpath
     dstpath = args.dstpath
-    prepare(srcpath, dstpath)
\ No newline at end of file
+    prepare(srcpath, dstpath)
diff --git a/configs/DOTA/faster_rcnn_obb_r50_fpn_1x_dota.py b/configs/DOTA/faster_rcnn_obb_r50_fpn_1x_dota.py
index d43c4ea..4d73e81 100644
--- a/configs/DOTA/faster_rcnn_obb_r50_fpn_1x_dota.py
+++ b/configs/DOTA/faster_rcnn_obb_r50_fpn_1x_dota.py
@@ -102,7 +102,7 @@
 )
 # dataset settings
 dataset_type = 'DOTADataset'
-data_root = 'data/dota1_1024/'
+data_root = '/content/AerialDetection/data/dota1-split-1024/'
 img_norm_cfg = dict(
     mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
 data = dict(