Add some landmarks HTML visualizations

Austin Blanton
1 parent 34a914c4
Showing 1 changed file with 35 additions and 2 deletions
scripts/brpy/html_viz.py
@@ -8,12 +8,46 @@ and host them as long as the relative paths remain the same on ya serva.
 from PIL import Image
+LMSIZE = 8
+
+def landmarks_on_crop(landmarks, x, y, width, height, imname, maxheight=None, color='green'):
+    '''
+    Generates an HTML string that shows landmarks within a given cropped image.
+    When two landmarked crops are put next to each other, they will be inline. To make each crop its own line, wrap it in a div.
+    '''
+    html = _bbcrop(x, y, width, height, imname, maxheight)
+    if not maxheight:
+        maxheight = height
+    ratio = float(maxheight) / height
+    print(ratio)
+    if len(landmarks) > 0 and len(landmarks[0]) != 3:
+        landmarks = [ lm + (color,) for lm in landmarks ]
+    for lmx,lmy,col in landmarks:
+        html += landmark((lmx-x)*ratio - (LMSIZE/2), (lmy-y)*ratio - (LMSIZE/2), col)
+    html += '</div>'
+    return html
+
+def landmark_img(x, y, img, color='green'):
+    html = '<div style="position:relative;">'
+    html += '<img src="%s" />' % img
+    html += landmark(x,y,color)
+    html += '</div>'
+    return html
+
+def landmark(x, y, color='green'):
+    return '<div style="position:absolute; top:{0}px; left:{1}px; color:{2}; background-color:{2}; width:{3}px; height:{3}px;"></div>'.format(y,x,color,LMSIZE)
+
 def crop_to_bb(x, y, width, height, imname, maxheight=None):
     '''
     Generates an HTML string that crops to a given bounding box and resizes to maxheight pixels.
     A maxheight of None will keep the original size (default).
     When two crops are put next to each other, they will be inline. To make each crop its own line, wrap it in a div.
     '''
+    html = _bbcrop(x, y, width, height, imname, maxheight)
+    html += '</div>'
+    return html
+
+def _bbcrop(x, y, width, height, imname, maxheight):
     img = Image.open(imname)
     imwidth, imheight = img.size
     if not maxheight:
@@ -23,9 +57,8 @@ def crop_to_bb(x, y, width, height, imname, maxheight=None):
     # image is cropped with div width/height + overflow:hidden,
     # resized with img height,
     # and positioned with img margin
-    html = '<div style="overflow:hidden; display:inline-block; width:%ipx; height:%ipx;">' % (width*ratio, maxheight)
+    html = '<div style="overflow:hidden; display:inline-block; position:relative; width:%ipx; height:%ipx;">' % (width*ratio, maxheight)
     html += '<img src="%s" style="height:%ipx; margin:-%ipx 0 0 -%ipx;"/>' % (imname, imheight*ratio, y*ratio, x*ratio)
-    html += '</div>'
     return html
 def bbs_for_image(imname, bbs, maxheight=None, colors=None):