上传主程序

draw.py

@@ -0,0 +1,114 @@
+import tensorflow as tf
+import numpy as np
+import random
+import pickle
+from PIL import Image
+from qrcode import make as makeqr
+from dnnlib import tflib
+import time, os, hashlib
+
+
+def main():
+        # Define global variables.
+        seed = random.randint(0,10000000)
+        available_charaters = {'Anmicius', 'Camil', 'Grey', 'King', 'Ray'}#这里是目前支持的角色列表
+		#Anmicius安迷修，Camil卡米尔，Grey格瑞，King金，Ray雷狮
+
+        # Select charater and input seed.
+        selected_character = 'Ray'#这里是选择角色，可以从上方复制
+        while selected_character not in available_charaters:
+            selected_character = input('Type in the character you want to draw, e.g. \"Anmicius\" and \"Ray\" (no quotes).\n')
+            if selected_character not in available_charaters:
+                print('You typed in a character that is not available or you made a misspell, try agian.')
+    
+        seed_str = ''
+        if seed_str != '':
+            if seed_str.isdigit():
+                seed = int(seed_str.encode('utf-8'))
+            else:
+                seed = int(hashlib.sha256(seed_str.encode('utf-8')).hexdigest(), 16) % 10**8
+    
+        print('INFO: Setting up variables...')
+        tflib.init_tf()
+        rnd = np.random.RandomState(seed)
+        fmt = dict(func=tflib.convert_images_to_uint8, nchw_to_nhwc=True)
+
+        print('INFO: Loading pretrained model...')
+        Gs = pickle.load(open('models/network-%s-gs.pkl' % selected_character, 'rb'))
+        latents = rnd.randn(1, Gs.input_shape[1])
+    
+        print('INFO: Generating...')
+        images = Gs.run(latents, None, truncation_psi=0.7, randomize_noise=True, output_transform=fmt)
+    
+        im = Image.fromarray(images[0], 'RGB')
+        qr = makeqr('This is an image automatically generated by Aotu Draw Bot by Rand0mZ.LiCloud provides computing resources. Seed: %d' % seed)
+        w, h = im.size
+        qw, qh = qr.size
+        if qw > w:
+            qr = qr.resize((w, w))
+        elif qh > h:
+            qr = qr.resize((h, h))
+        qw, qh = qr.size
+
+        imd = im.load()
+        for i in range(w):
+            for j in range(h):
+                d = imd[i, j]
+                imd[i, j] = d[:-1] +((d[-1] | 1) if qr.getpixel((i%qw, j%qh)) else (d[-1] & ~1),)
+
+        print('Done!')
+
+        save_name = '%s_%d.png' % (selected_character, seed)
+        print('INFO: Saving %s' % save_name)
+        output_dir = os.path.join(os.getcwd() ,'arags')#这里是保存目录，默认保存到arags目录下
+        if not os.path.isdir(output_dir):
+            os.mkdir(output_dir)
+        im.save(os.path.join(output_dir, save_name))
+        print('INFO: Image %s is saved in directory .' % save_name)
+        print('INFO: All processes has done!')
+        print('Thank you for using this software and obeying the terms of use above.')
+        time.sleep(3)
+        
+
+def generate_image(model, save_path, selected_character, seed, amount):
+    tflib.init_tf()
+
+    print('INFO: Loading pretrained model...')
+    Gs = pickle.load(open(model, 'rb'))
+
+    if not os.path.isdir(save_path):
+        os.mkdir(save_path)
+
+    for i in range(1, amount + 1):
+        print('INFO: Generating image %d' %i)
+        rnd = np.random.RandomState(seed)
+        fmt = dict(func=tflib.convert_images_to_uint8, nchw_to_nhwc=True)
+        latents = rnd.randn(1, Gs.input_shape[1])
+        images = Gs.run(latents, None, truncation_psi=0.7, randomize_noise=True, output_transform=fmt)
+
+        im = Image.fromarray(images[0], 'RGB')
+        qr = makeqr('This is an image automatically generated by Aotu Draw Bot CLI by Rand0mZ hence this image is not for commercial propose. Seed: %d' % seed)
+        w, h = im.size
+        qw, qh = qr.size
+        if qw > w:
+            qr = qr.resize((w, w))
+        elif qh > h:
+            qr = qr.resize((h, h))
+        qw, qh = qr.size
+
+        imd = im.load()
+        for i in range(w):
+            for j in range(h):
+                d = imd[i, j]
+                imd[i, j] = d[:-1] +((d[-1] | 1) if qr.getpixel((i%qw, j%qh)) else (d[-1] & ~1),)
+
+        print('Done!')
+
+        save_name = '%s_%d.png' % (selected_character, seed)
+        print('INFO: Saving %s' % save_name)
+        im.save(os.path.join(save_path, save_name))
+        print('INFO: Image %s is saved to %s.\n' % (save_name, save_path))
+        seed += i - 1
+
+if __name__ == "__main__":
+    main()

redo.py

@@ -0,0 +1,5 @@
+import os
+
+for i in range(1,1001):#运行次数 可以按照需要把第二个数字改成你想要的运行次数+1，默认1000次
+    os.system('py draw.py')
+#这个是重复运行的脚本，单次运行只需要运行draw.py即可