Merged PR 3142: Million Scale Python Sample Update

Code format & update for sample.

Author: Xuan (Sean) Hu

sample/__main__.py

@@ -7,7 +7,6 @@
 
 from view import MyApp
 
-
 if __name__ == "__main__":
     app = MyApp(False)
     app.MainLoop()

sample/model/face.py

@@ -11,6 +11,7 @@
 
 class Rect(object):
     """Face Rectangle."""
+
     def __init__(self, rect):
         super(Rect, self).__init__()
         self.set_rect(rect)
@@ -25,6 +26,7 @@ def set_rect(self, rect):
 
 class Attribute(object):
     """Attributes for face."""
+
     def __init__(self, attr):
         super(Attribute, self).__init__()
         self.set_attr(attr)
@@ -41,29 +43,26 @@ def set_attr(self, attr):
         else:
             self.hair = max(
                 attr['hair']['hairColor'],
-                key=lambda x: x['confidence']
-            )['color']
+                key=lambda x: x['confidence'])['color']
         self.facial_hair = sum(attr['facialHair'].values()) > 0 and 'Yes' \
             or 'No'
         self.makeup = any(attr['makeup'].values())
         self.emotion = util.key_with_max_value(attr['emotion'])
         self.occlusion = any(attr['occlusion'].values())
         self.exposure = attr['exposure']['exposureLevel']
         self.head_pose = "Pitch: {}, Roll:{}, Yaw:{}".format(
-            attr['headPose']['pitch'],
-            attr['headPose']['roll'],
-            attr['headPose']['yaw']
-        )
+            attr['headPose']['pitch'], attr['headPose']['roll'],
+            attr['headPose']['yaw'])
         if not attr['accessories']:
             self.accessories = 'NoAccessories'
         else:
             self.accessories = ' '.join(
-                [str(x['type']) for x in attr['accessories']]
-            )
+                [str(x['type']) for x in attr['accessories']])
 
 
 class Face(object):
     """Face Model for each face."""
+
     def __init__(self, res, path, size=util.MAX_THUMBNAIL_SIZE):
         super(Face, self).__init__()
         self.path = path
@@ -76,18 +75,13 @@ def __init__(self, res, path, size=util.MAX_THUMBNAIL_SIZE):
             self.persisted_id = res['persistedFaceId']
         if res.get('faceRectangle'):
             self.rect = Rect(res['faceRectangle'])
-            self.bmp = self.bmp.GetSubBitmap(wx.Rect(
-                self.rect.left,
-                self.rect.top,
-                self.rect.width,
-                self.rect.height,
-            ))
+            self.bmp = self.bmp.GetSubBitmap(
+                wx.Rect(self.rect.left, self.rect.top, self.rect.width,
+                        self.rect.height))
         if res.get('faceAttributes'):
             self.attr = Attribute(res['faceAttributes'])
         self.bmp = util.scale_image(
-            self.bmp.ConvertToImage(),
-            size=size,
-        ).ConvertToBitmap()
+            self.bmp.ConvertToImage(), size=size).ConvertToBitmap()
 
     def set_name(self, name):
         """Set the name for the face."""

sample/util.py

@@ -20,7 +20,6 @@
     sys.path.insert(0, ROOT_DIR)
     import cognitive_face as CF
 
-
 IMAGE_WILDCARD = 'Image files (*.jpg, *.png)|*.jpg; *.png'
 INNER_PANEL_WIDTH = 710
 MAX_IMAGE_SIZE = 300
@@ -32,17 +31,14 @@
 
 LOG_FACE_LIST_REQUEST = (
     'Request: Face List {} will be used for build person database. '
-    'Checking whether group exists.'
-)
+    'Checking whether group exists.')
 LOG_FACE_LIST_NOT_EXIST = 'Response: Face List {} does not exist before.'
 LOG_FACE_LIST_EXIST = 'Response: Face List {} exists.'
-LABEL_FACE = (
-    '{}, {} years old\n'
-    'Hair: {}, Facial Hair: {}\n'
-    'Makeup: {}, Emotion: {}\n'
-    'Occluded: {}, Exposure: {}\n'
-    '{}\n{}\n'
-)
+LABEL_FACE = ('{}, {} years old\n'
+              'Hair: {}, Facial Hair: {}\n'
+              'Makeup: {}, Emotion: {}\n'
+              'Occluded: {}, Exposure: {}\n'
+              '{}\n{}\n')
 
 
 class SubscriptionKey(object):
@@ -67,7 +63,7 @@ def set(cls, key):
         """Set the subscription key."""
         cls.key = key
         with file(SUBSCRIPTION_KEY_FILENAME, 'w') as fout:
-            print >>fout, key
+            print >> fout, key
         CF.Key.set(cls.key)
 
     @classmethod
@@ -101,7 +97,7 @@ def set(cls, endpoint):
         """Set the endpoint."""
         cls.endpoint = endpoint
         with file(ENDPOINT_FILENAME, 'w') as fout:
-            print >>fout, endpoint
+            print >> fout, endpoint
         CF.BaseUrl.set(cls.endpoint)
 
     @classmethod
@@ -151,21 +147,16 @@ def draw_bitmap_rectangle(bitmap, faces):
     dc.SetTextBackground('black')
     dc.SetTextForeground('white')
     dc.SetBackgroundMode(wx.SOLID)
-    dc.SetFont(wx.Font(8,
-                       wx.FONTFAMILY_DEFAULT,
-                       wx.FONTSTYLE_NORMAL,
-                       wx.FONTWEIGHT_BOLD))
+    dc.SetFont(
+        wx.Font(8, wx.FONTFAMILY_DEFAULT, wx.FONTSTYLE_NORMAL,
+                wx.FONTWEIGHT_BOLD))
     for face in faces:
         dc.DrawRectangle(
-            face.rect.left * bitmap.scale,
-            face.rect.top * bitmap.scale,
-            face.rect.width * bitmap.scale,
-            face.rect.height * bitmap.scale,
-        )
+            face.rect.left * bitmap.scale, face.rect.top * bitmap.scale,
+            face.rect.width * bitmap.scale, face.rect.height * bitmap.scale)
         if face.name:
             text_width, text_height = dc.GetTextExtent(face.name)
-            dc.DrawText(face.name,
-                        face.rect.left * bitmap.scale,
+            dc.DrawText(face.name, face.rect.left * bitmap.scale,
                         face.rect.top * bitmap.scale - text_height)
     dc.SelectObject(wx.NullBitmap)
     bitmap.bitmap.SetBitmap(bitmap.bmp)
@@ -185,8 +176,10 @@ def key_with_max_value(item):
 
 def async(func):
     """Async wrapper."""
+
     def wrapper(*args, **kwargs):
         """docstring for wrapper"""
         thr = Thread(target=func, args=args, kwargs=kwargs)
         thr.start()
+
     return wrapper

sample/view/__init__.py

@@ -24,6 +24,7 @@
 
 class MyLabelBook(LB.LabelBook):
     """LabelBook part in Main Frame."""
+
     def __init__(self, parent):
         agw_style = INB_LEFT | INB_FIT_LABELTEXT | INB_NO_RESIZE
         super(MyLabelBook, self).__init__(parent, agwStyle=agw_style)
@@ -44,6 +45,7 @@ def __init__(self, parent):
 
 class MyTitle(wx.Panel):
     """Title part in Main Frame."""
+
     def __init__(self, parent):
         super(MyTitle, self).__init__(parent)
         self.SetBackgroundColour('#00b294')
@@ -66,6 +68,7 @@ def __init__(self, parent):
 
 class MyFrame(wx.Frame):
     """Main Frame."""
+
     def __init__(self, parent):
         super(MyFrame, self).__init__(parent, title=TITLE, size=(1280, 768))
 
@@ -83,8 +86,7 @@ def __init__(self, parent):
         status_text = (
             'Microsoft will receive the images you upload and may use them to '
             'improve Face API and related services. By submitting an image, '
-            'you confirm you have consent from everyone in it.'
-        )
+            'you confirm you have consent from everyone in it.')
         self.status = wx.StatusBar(self)
         self.status.SetStatusText(status_text)
         sizer.Add(self.status, flag=wx.EXPAND)
@@ -95,6 +97,7 @@ def __init__(self, parent):
 
 class MyApp(wx.App):
     """The whole app."""
+
     def OnInit(self):
         """Show main frame."""
         frame = MyFrame(None)

sample/view/base.py

@@ -13,6 +13,7 @@
 
 class MyPanel(wx.Panel):
     """Base Panel."""
+
     def __init__(self, parent):
         super(MyPanel, self).__init__(parent)
         colour_window = wx.SystemSettings.GetColour(wx.SYS_COLOUR_WINDOW)
@@ -21,6 +22,7 @@ def __init__(self, parent):
 
 class MyStaticBitmap(MyPanel):
     """Base StaticBitmap."""
+
     def __init__(self, parent, bitmap=wx.NullBitmap, size=util.MAX_IMAGE_SIZE):
         super(MyStaticBitmap, self).__init__(parent)
         self.bmp = bitmap
@@ -49,7 +51,13 @@ def set_path(self, path):
 
 class MyGridStaticBitmap(wx.Panel):
     """Base Grid StaticBitmap."""
-    def __init__(self, parent, rows=1, cols=0, vgap=0, hgap=0,
+
+    def __init__(self,
+                 parent,
+                 rows=1,
+                 cols=0,
+                 vgap=0,
+                 hgap=0,
                  size=util.MAX_THUMBNAIL_SIZE):
         super(MyGridStaticBitmap, self).__init__(parent)
         self.sizer = wx.GridSizer(rows, cols, vgap, hgap)
@@ -78,6 +86,7 @@ def set_faces(self, faces):
 
 class WrapCaptionFaceList(wx.WrapSizer):
     """Wrap face list with caption under the face."""
+
     def __init__(self, parent, confidence_faces, size=util.MAX_THUMBNAIL_SIZE):
         super(WrapCaptionFaceList, self).__init__()
         for face, confidence in confidence_faces:
@@ -96,6 +105,7 @@ def __init__(self, parent, confidence_faces, size=util.MAX_THUMBNAIL_SIZE):
 
 class FindSimilarsResult(wx.Panel):
     """The view for Find Similar result."""
+
     def __init__(self, parent):
         super(FindSimilarsResult, self).__init__(parent)
         self.sizer = wx.BoxSizer(wx.VERTICAL)
@@ -104,7 +114,8 @@ def set_data(self, faces, res_tot, size=util.MAX_THUMBNAIL_SIZE):
         """Set the data."""
         self.sizer.Clear(True)
         static_text_title = wx.StaticText(
-            self, label='Find {} Similar Candidate Faces Results:'.format(
+            self,
+            label='Find {} Similar Candidate Faces Results:'.format(
                 len(faces)))
         self.sizer.Add(static_text_title, 0, wx.EXPAND)
 
@@ -133,6 +144,7 @@ def set_data(self, faces, res_tot, size=util.MAX_THUMBNAIL_SIZE):
 
 class WrapFaceList(wx.Panel):
     """Base wrap face list."""
+
     def __init__(self, parent, faces, size=util.MAX_THUMBNAIL_SIZE):
         super(WrapFaceList, self).__init__(parent)
         self.sizer = wx.WrapSizer()
@@ -146,6 +158,7 @@ def __init__(self, parent, faces, size=util.MAX_THUMBNAIL_SIZE):
 
 class CaptionWrapFaceList(wx.Panel):
     """Wrap face list with a caption."""
+
     def __init__(self, parent):
         super(CaptionWrapFaceList, self).__init__(parent)
         self.sizer = wx.BoxSizer(wx.VERTICAL)
@@ -165,6 +178,7 @@ def set_data(self, caption_faces_list, size=util.MAX_THUMBNAIL_SIZE):
 
 class GroupResult(wx.Panel):
     """The view for Group result."""
+
     def __init__(self, parent):
         super(GroupResult, self).__init__(parent)
         self.sizer = wx.BoxSizer(wx.VERTICAL)
@@ -195,6 +209,7 @@ def set_data(self, faces, res, size=util.MAX_THUMBNAIL_SIZE):
 
 class MyLog(wx.TextCtrl):
     """The window for each scenario."""
+
     def __init__(self, parent):
         style = wx.TE_MULTILINE | wx.TE_READONLY
         super(MyLog, self).__init__(parent, style=style)
@@ -210,6 +225,7 @@ def log(self, msg):
 
 class MyFaceList(wx.VListBox):
     """Face List."""
+
     def __init__(self, parent, faces=[], **kwargs):
         super(MyFaceList, self).__init__(parent, **kwargs)
         self.SetItems(faces)
@@ -230,17 +246,10 @@ def OnDrawItem(self, dc, rect, index):
         textx = rect.x + 2 + face.bmp.GetWidth() + 2
         label_rect = wx.Rect(textx, rect.y, rect.width - textx, rect.height)
         label = util.LABEL_FACE.format(
-            face.attr.gender,
-            face.attr.age,
-            face.attr.hair,
-            face.attr.facial_hair,
-            face.attr.makeup,
-            face.attr.emotion,
-            face.attr.occlusion,
-            face.attr.exposure,
-            face.attr.head_pose,
-            face.attr.accessories
-        )
+            face.attr.gender, face.attr.age, face.attr.hair,
+            face.attr.facial_hair, face.attr.makeup, face.attr.emotion,
+            face.attr.occlusion, face.attr.exposure, face.attr.head_pose,
+            face.attr.accessories)
         dc.DrawLabel(label, label_rect, wx.ALIGN_LEFT | wx.ALIGN_TOP)
 
     def SetItems(self, faces):

sample/view/panel_detection.py

@@ -14,6 +14,7 @@
 
 class DetectionPanel(base.MyPanel):
     """Detection Panel."""
+
     def __init__(self, parent):
         super(DetectionPanel, self).__init__(parent)
 
@@ -25,12 +26,10 @@ def __init__(self, parent):
         self.hvsizer = wx.BoxSizer(wx.VERTICAL)
         self.hvsizer.SetMinSize((util.INNER_PANEL_WIDTH, -1))
 
-        label = (
-            "To detect faces in an image, click the 'Choose Image' "
-            "button. You will see a rectangle surrounding every face "
-            "that the Face API detects. You will also see a list of "
-            "attributes related to the faces."
-        )
+        label = ("To detect faces in an image, click the 'Choose Image' "
+                 "button. You will see a rectangle surrounding every face "
+                 "that the Face API detects. You will also see a list of "
+                 "attributes related to the faces.")
         self.static_text = wx.StaticText(self, label=label)
         self.static_text.Wrap(util.INNER_PANEL_WIDTH)
         self.hvsizer.Add(self.static_text, 0, wx.ALL, 5)
@@ -102,8 +101,7 @@ def async_detect(self, path):
         try:
             attributes = (
                 'age,gender,headPose,smile,facialHair,glasses,emotion,hair,'
-                'makeup,occlusion,accessories,blur,exposure,noise'
-            )
+                'makeup,occlusion,accessories,blur,exposure,noise')
             res = util.CF.face.detect(path, False, False, attributes)
             faces = [model.Face(face, path) for face in res]
             self.face_list.SetItems(faces)