fix get_sbd

Author: knorth55

chainercv/datasets/sbd/sbd_utils.py

@@ -10,27 +10,7 @@
 train_voc2012_url = 'http://home.bharathh.info/pubs/codes/SBD/train_noval.txt'
 
 
-def get_sbd():
-    data_root = download.get_dataset_directory(root)
-    base_path = os.path.join(data_root, 'benchmark_RELEASE/dataset')
-    train_voc2012_file = os.path.join(base_path, 'train_voc2012.txt')
-    if not os.path.exists(train_voc2012_file):
-        six.moves.urllib.request.urlretrieve(
-            train_voc2012_url, train_voc2012_file)
-
-    if os.path.exists(os.path.join(base_path, 'train.txt')):
-        # skip downloading
-        return base_path
-
-    download_file_path = utils.cached_download(url)
-    ext = os.path.splitext(url)[1]
-    utils.extractall(download_file_path, data_root, ext)
-    generate_voc2012_txt(base_path)
-
-    return base_path
-
-
-def generate_voc2012_txt(base_path):
+def _generate_voc2012_txt(base_path):
     with open(os.path.join(base_path, 'train.txt'), 'r') as f:
         train_ids = f.read().split('\n')[:-1]
     with open(os.path.join(base_path, 'val.txt'), 'r') as f:
@@ -46,6 +26,25 @@ def generate_voc2012_txt(base_path):
         f.write('\n'.join(sorted(all_ids)))
 
 
+def get_sbd():
+    data_root = download.get_dataset_directory(root)
+    base_path = os.path.join(data_root, 'benchmark_RELEASE/dataset')
+
+    train_voc2012_file = os.path.join(base_path, 'train_voc2012.txt')
+    if os.path.exists(train_voc2012_file):
+        # skip downloading
+        return base_path
+
+    download_file_path = utils.cached_download(url)
+    ext = os.path.splitext(url)[1]
+    utils.extractall(download_file_path, data_root, ext)
+
+    six.moves.urllib.request.urlretrieve(train_voc2012_url, train_voc2012_file)
+    _generate_voc2012_txt(base_path)
+
+    return base_path
+
+
 sbd_instance_segmentation_label_names = (
     'background',
     'aeroplane',