cleanup and code formatting

Author: adityaarun1

lib/datasets/coco.py

@@ -11,39 +11,39 @@
 
 
 def unique_boxes(boxes, scale=1.0):
-  """Return indices of unique boxes."""
-  v = np.array([1, 1e3, 1e6, 1e9])
-  hashes = np.round(boxes * scale).dot(v)
-  _, index = np.unique(hashes, return_index=True)
-  return np.sort(index)
+    """Return indices of unique boxes."""
+    v = np.array([1, 1e3, 1e6, 1e9])
+    hashes = np.round(boxes * scale).dot(v)
+    _, index = np.unique(hashes, return_index=True)
+    return np.sort(index)
 
 
 def xywh_to_xyxy(boxes):
-  """Convert [x y w h] box format to [x1 y1 x2 y2] format."""
-  return np.hstack((boxes[:, 0:2], boxes[:, 0:2] + boxes[:, 2:4] - 1))
+    """Convert [x y w h] box format to [x1 y1 x2 y2] format."""
+    return np.hstack((boxes[:, 0:2], boxes[:, 0:2] + boxes[:, 2:4] - 1))
 
 
 def xyxy_to_xywh(boxes):
-  """Convert [x1 y1 x2 y2] box format to [x y w h] format."""
-  return np.hstack((boxes[:, 0:2], boxes[:, 2:4] - boxes[:, 0:2] + 1))
+    """Convert [x1 y1 x2 y2] box format to [x y w h] format."""
+    return np.hstack((boxes[:, 0:2], boxes[:, 2:4] - boxes[:, 0:2] + 1))
 
 
 def validate_boxes(boxes, width=0, height=0):
-  """Check that a set of boxes are valid."""
-  x1 = boxes[:, 0]
-  y1 = boxes[:, 1]
-  x2 = boxes[:, 2]
-  y2 = boxes[:, 3]
-  assert (x1 >= 0).all()
-  assert (y1 >= 0).all()
-  assert (x2 >= x1).all()
-  assert (y2 >= y1).all()
-  assert (x2 < width).all()
-  assert (y2 < height).all()
+    """Check that a set of boxes are valid."""
+    x1 = boxes[:, 0]
+    y1 = boxes[:, 1]
+    x2 = boxes[:, 2]
+    y2 = boxes[:, 3]
+    assert (x1 >= 0).all()
+    assert (y1 >= 0).all()
+    assert (x2 >= x1).all()
+    assert (y2 >= y1).all()
+    assert (x2 < width).all()
+    assert (y2 < height).all()
 
 
 def filter_small_boxes(boxes, min_size):
-  w = boxes[:, 2] - boxes[:, 0]
-  h = boxes[:, 3] - boxes[:, 1]
-  keep = np.where((w >= min_size) & (h > min_size))[0]
-  return keep
+    w = boxes[:, 2] - boxes[:, 0]
+    h = boxes[:, 3] - boxes[:, 1]
+    keep = np.where((w >= min_size) & (h > min_size))[0]
+    return keep

lib/datasets/ds_utils.py

@@ -4,7 +4,6 @@
 # Licensed under The MIT License [see LICENSE for details]
 # Written by Ross Girshick
 # --------------------------------------------------------
-
 """Factory method for easily getting imdbs by name."""
 from __future__ import absolute_import
 from __future__ import division
@@ -16,37 +15,37 @@
 
 import numpy as np
 
-# Set up voc_<year>_<split> 
+# Set up voc_<year>_<split>
 for year in ['2007', '2012']:
-  for split in ['train', 'val', 'trainval', 'test']:
-    name = 'voc_{}_{}'.format(year, split)
-    __sets[name] = (lambda split=split, year=year: pascal_voc(split, year))
+    for split in ['train', 'val', 'trainval', 'test']:
+        name = 'voc_{}_{}'.format(year, split)
+        __sets[name] = (lambda split=split, year=year: pascal_voc(split, year))
 
 for year in ['2007', '2012']:
-  for split in ['train', 'val', 'trainval', 'test']:
-    name = 'voc_{}_{}_diff'.format(year, split)
-    __sets[name] = (lambda split=split, year=year: pascal_voc(split, year, use_diff=True))
+    for split in ['train', 'val', 'trainval', 'test']:
+        name = 'voc_{}_{}_diff'.format(year, split)
+        __sets[name] = (lambda split=split, year=year: pascal_voc(split, year, use_diff=True))
 
 # Set up coco_2014_<split>
 for year in ['2014']:
-  for split in ['train', 'val', 'minival', 'valminusminival', 'trainval']:
-    name = 'coco_{}_{}'.format(year, split)
-    __sets[name] = (lambda split=split, year=year: coco(split, year))
+    for split in ['train', 'val', 'minival', 'valminusminival', 'trainval']:
+        name = 'coco_{}_{}'.format(year, split)
+        __sets[name] = (lambda split=split, year=year: coco(split, year))
 
 # Set up coco_2015_<split>
 for year in ['2015']:
-  for split in ['test', 'test-dev']:
-    name = 'coco_{}_{}'.format(year, split)
-    __sets[name] = (lambda split=split, year=year: coco(split, year))
+    for split in ['test', 'test-dev']:
+        name = 'coco_{}_{}'.format(year, split)
+        __sets[name] = (lambda split=split, year=year: coco(split, year))
 
 
 def get_imdb(name):
-  """Get an imdb (image database) by name."""
-  if name not in __sets:
-    raise KeyError('Unknown dataset: {}'.format(name))
-  return __sets[name]()
+    """Get an imdb (image database) by name."""
+    if name not in __sets:
+        raise KeyError('Unknown dataset: {}'.format(name))
+    return __sets[name]()
 
 
 def list_imdbs():
-  """List all registered imdbs."""
-  return list(__sets.keys())
+    """List all registered imdbs."""
+    return list(__sets.keys())