update to 0.2.1

Author: twmht

README.md

@@ -3,13 +3,10 @@ Simple WIDERFACE data parser written in python
 
 WIDERFACE is a database of face bounding boxes available on http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/.
 
-You can download the database via `python get_data.py`
-
 ## Requirement
-h5py(v2.6.0)
+scipy(v0.19.0)
 
 ## Usage
 Please see `wider/examples/example.py` for usage
 
 you also need opencv and matplotlib (v2.0.0) to run the example.
-

requirement.txt

@@ -1 +1 @@
-h5py==2.6.0
+scipy==0.19.0

setup.py

@@ -6,7 +6,7 @@
     from distutils.core import setup
 
 NAME = 'python-widerface'
-VERSION = '0.1.2'
+VERSION = '0.2.1'
 
 # Compile the list of packages available, because distutils doesn't have
 # an easy way to do this.

wider/examples/example.py

@@ -2,9 +2,12 @@
 import matplotlib.pyplot as plt
 import cv2
 
-wider = WIDER('/home/tumh/python-wider/data/v1',
-              '/home/tumh/python-wider/data/WIDER_train/images',
-              'wider_face_train.mat')
+# arg1: path to label
+# arg2: path to images
+# arg3: label file name
+wider = WIDER('/home/tumh/wider_face_split',
+              '/home/tumh/WIDER_val/images',
+              'wider_face_val.mat')
 
 
 # press ctrl-C to stop the process

wider/loader.py

@@ -1,5 +1,6 @@
-import h5py
+import matplotlib.pyplot as plt
 import os
+import scipy.io
 
 
 class DATA(object):
@@ -9,33 +10,43 @@ def __init__(self, image_name, bboxes):
 
 
 class WIDER(object):
+    """
+    Build a wider parser
+
+    Parameters
+    ----------
+    path_to_label : path of the label file
+    path_to_image : path of the image files
+    fname : name of the label file
+
+    Returns
+    -------
+    a wider parser
+    """
     def __init__(self, path_to_label, path_to_image, fname):
         self.path_to_label = path_to_label
         self.path_to_image = path_to_image
 
-        self.f = h5py.File(os.path.join(path_to_label, fname), 'r')
+        self.f = scipy.io.loadmat(os.path.join(path_to_label, fname))
         self.event_list = self.f.get('event_list')
         self.file_list = self.f.get('file_list')
         self.face_bbx_list = self.f.get('face_bbx_list')
 
     def next(self):
-
-        for event_idx, event in enumerate(self.event_list.value[0]):
-            directory = self.f[event].value.tostring().decode('utf-16')
-            for im_idx, im in enumerate(
-                    self.f[self.file_list.value[0][event_idx]].value[0]):
-
-                im_name = self.f[im].value.tostring().decode('utf-16')
-                face_bbx = self.f[self.f[self.face_bbx_list.value
-                                  [0][event_idx]].value[0][im_idx]].value
+        for event_idx, event in enumerate(self.event_list):
+            directory = event[0][0]
+            for im_idx, im in enumerate(self.file_list[event_idx][0]):
+                im_name = im[0][0]
+                face_bbx = self.face_bbx_list[event_idx][0][im_idx][0]
+                #  print face_bbx.shape
 
                 bboxes = []
 
-                for i in range(face_bbx.shape[1]):
-                    xmin = int(face_bbx[0][i])
-                    ymin = int(face_bbx[1][i])
-                    xmax = int(face_bbx[0][i] + face_bbx[2][i])
-                    ymax = int(face_bbx[1][i] + face_bbx[3][i])
+                for i in range(face_bbx.shape[0]):
+                    xmin = int(face_bbx[i][0])
+                    ymin = int(face_bbx[i][1])
+                    xmax = int(face_bbx[i][2]) + xmin
+                    ymax = int(face_bbx[i][3]) + ymin
                     bboxes.append((xmin, ymin, xmax, ymax))
 
                 yield DATA(os.path.join(self.path_to_image, directory,