原始程式:https://github.com/kosmosmo/NFT_Art_Generator
修訂的程式:黃底是有修訂部份,126-130行是把註解拿掉。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 | project_folder = '<您的目錄>/NFT_Art_Generator-master/example/' total_art = 100 from collections import OrderedDict import os, json, random,cv2 import numpy as np def create_json(): layers_map = OrderedDict() my_layers = os.listdir(project_folder+'layers') for layer in my_layers: layers_map[layer] = {} pics = os.listdir(project_folder+'layers'+ '\\' + layer) temp = {'0_no_attributes':{'att_chance':1.0,'black_list_att':[]}} for p in pics: temp[p.split('.')[0]] = {'att_chance':1.0,'black_list_att':[]} layers_map[layer]['link_att'] = None layers_map[layer]['attributes'] = temp with open(project_folder + 'layers_map.json', 'w') as f: f.write(json.dumps(layers_map, indent=4, separators=(',', ': '), sort_keys=True)) return layers_map def get_weight(attributes): population = [] weight = [] for key,val in attributes.items(): population.append(key) weight.append(val['att_chance']) return [population,weight] def generate_att(): with open(project_folder + 'layers_map.json', 'r') as f: layers_map = json.load(f, object_pairs_hook=OrderedDict) attribute_map = OrderedDict() visited = set() i = 0 while i < total_art: print ('generating number ' + str(i)) temp = OrderedDict() atts = '' for key,val in layers_map.items(): if val['link_att'] == None: population, weights = get_weight(val['attributes']) random_pick = random.choices(population=population, weights=weights)[0] temp[key] = random_pick else: if temp[val['link_att']] in val['attributes']: temp[key] = temp[val['link_att']] else: temp[key] ='0_no_attributes' atts += key + ':' + temp[key]+ ',' if atts not in visited: visited.add(atts) attribute_map[i] = temp i += 1 with open(project_folder + 'attributes_map.json', 'w') as f: f.write(json.dumps(attribute_map, indent=4, separators=(',', ': '), sort_keys=True)) return attribute_map def generate_art(attribute_map=None): if not attribute_map: with open(project_folder + 'attributes_map.json', 'r') as f: attribute_map = json.load(f, object_pairs_hook=OrderedDict) for key,val in attribute_map.items(): print('generating img ' + key) img1 = np.zeros((1, 1,3), dtype=np.uint8) for key1,val1 in val.items(): if val1 == "0_no_attributes":continue img_path = project_folder + 'layers\\' +key1 + '\\' + val1 + ".png" fg = cv2.imread(img_path, -1) fg = cv2.cvtColor(fg, cv2.COLOR_RGB2RGBA) fgMask = fg[:, :, 3:] img = fg[:, :, :-1] bgMask = 255 - fgMask imgMask = cv2.cvtColor(fgMask, cv2.COLOR_GRAY2BGR) bgMask = cv2.cvtColor(bgMask, cv2.COLOR_GRAY2BGR) bgNew = (img1 * (1 / 255.0)) * (bgMask * (1 / 255.0)) imgNew = (img * (1 / 255.0)) * (imgMask * (1 / 255.0)) result = np.uint8(cv2.addWeighted(bgNew, 255.0, imgNew, 255.0, 0.0)) img1 = result """ # no compression cv2.imwrite(project_folder + 'outputs\\'+ key +'.png', img1) """ cv2.imwrite(project_folder + 'outputs\\'+ key + '.jpg', img1, [int(cv2.IMWRITE_JPEG_QUALITY), 90]) def generate_trait_data(): with open(project_folder + 'attributes_map.json', 'r') as f: attribute_map = json.load(f, object_pairs_hook=OrderedDict) trait_data_path = project_folder + 'outputs_data/' for key,val in attribute_map.items(): res = [] for k,v in val.items(): if v == "0_no_attributes":continue att = {} k = '_'.join(k.split('_')[1:]) att['trait_type'] = k att['value'] = v.replace('_', ' ') res.append(att) with open(trait_data_path + str(key) +'.json', 'w') as f: f.write(json.dumps(res, indent=4, separators=(',', ': '), sort_keys=True)) def slicing_dic(order_dic,start,end): list_dict = list(order_dic.items()) slice_dic = list_dict[start:end] res = OrderedDict(slice_dic) return (res) ''' from itertools import islice from multiprocessing import Process if __name__ == '__main__': with open(project_folder + 'attributes_map.json', 'r') as f: attribute_map = json.load(f, object_pairs_hook=OrderedDict) #attribute_map = slicing_dic(attribute_map,0,2000) # breaking the map into smaller chunks for render. def chunks(data, SIZE=100): it = iter(data) for i in range(0, len(data), SIZE): yield {k:data[k] for k in islice(it, SIZE)} for item in chunks(attribute_map): proc = Process(target=generate_art, args=(item,)) proc.start() ''' create_json() generate_att() generate_art() generate_trait_data() |
3.因應第91行,另外還需要建立outputs_data的目錄。
4.若找不到cv2套件,可以執行pip install opencv-python
5.執行
python generator.py
6.查看結果:
圖像:
JSON資料檔:
打開0.json
[
{
"trait_type": "feet",
"value": "feet"
},
{
"trait_type": "cloth",
"value": "blue pant"
},
{
"trait_type": "shoe",
"value": "dress"
},
{
"trait_type": "item",
"value": "choco"
},
{
"trait_type": "hand",
"value": "hand"
},
{
"trait_type": "sleeve",
"value": "squidward"
},
{
"trait_type": "face",
"value": "nah"
},
{
"trait_type": "head",
"value": "soda hat"
},
{
"trait_type": "glasses",
"value": "heart glasses"
}
]
沒有留言:
張貼留言