formatting

modules/textual_inversion/image_embedding.py

@@ -6,38 +6,46 @@ from PIL import Image,PngImagePlugin,ImageDraw,ImageFont
 from fonts.ttf import Roboto
 import torch
 
+
 class EmbeddingEncoder(json.JSONEncoder):
     def default(self, obj):
         if isinstance(obj, torch.Tensor):
             return {'TORCHTENSOR': obj.cpu().detach().numpy().tolist()}
         return json.JSONEncoder.default(self, obj)
 
+
 class EmbeddingDecoder(json.JSONDecoder):
     def __init__(self, *args, **kwargs):
         json.JSONDecoder.__init__(self, object_hook=self.object_hook, *args, **kwargs)
+
     def object_hook(self, d):
         if 'TORCHTENSOR' in d:
             return torch.from_numpy(np.array(d['TORCHTENSOR']))
         return d
 
+
 def embedding_to_b64(data):
     d = json.dumps(data, cls=EmbeddingEncoder)
     return base64.b64encode(d.encode())
 
+
 def embedding_from_b64(data):
     d = base64.b64decode(data)
     return json.loads(d, cls=EmbeddingDecoder)
 
+
 def lcg(m=2**32, a=1664525, c=1013904223, seed=0):
     while True:
         seed = (a * seed + c) % m
         yield seed % 255
 
+
 def xor_block(block):
     g = lcg()
     randblock = np.array([next(g) for _ in range(np.product(block.shape))]).astype(np.uint8).reshape(block.shape)
     return np.bitwise_xor(block.astype(np.uint8), randblock & 0x0F)
 
+
 def style_block(block, sequence):
     im = Image.new('RGB', (block.shape[1], block.shape[0]))
     draw = ImageDraw.Draw(im)
@@ -55,6 +63,7 @@ def style_block(block,sequence):
 
     return block ^ fg
 
+
 def insert_image_data_embed(image, data):
     d = 3
     data_compressed = zlib.compress(json.dumps(data, cls=EmbeddingEncoder).encode(), level=9)
@@ -90,6 +99,7 @@ def insert_image_data_embed(image,data):
 
     return background
 
+
 def crop_black(img, tol=0):
     mask = (img > tol).all(2)
     mask0, mask1 = mask.any(0), mask.any(1)
@@ -97,6 +107,7 @@ def crop_black(img,tol=0):
     row_start, row_end = mask1.argmax(), mask.shape[0]-mask1[::-1].argmax()
     return img[row_start:row_end, col_start:col_end]
 
+
 def extract_image_data_embed(image):
     d = 3
     outarr = crop_black(np.array(image.convert('RGB').getdata()).reshape(image.size[1], image.size[0], d).astype(np.uint8)) & 0x0F
@@ -117,6 +128,7 @@ def extract_image_data_embed(image):
     data = zlib.decompress(data_block)
     return json.loads(data, cls=EmbeddingDecoder)
 
+
 def caption_image_overlay(srcimage, title, footerLeft, footerMid, footerRight, textfont=None):
     from math import cos
 
@@ -163,10 +175,10 @@ def caption_image_overlay(srcimage,title,footerLeft,footerMid,footerRight,textfo
 
     return image
 
+
 if __name__ == '__main__':
 
     testEmbed = Image.open('test_embedding.png')
-    
     data = extract_image_data_embed(testEmbed)
     assert data is not None