19 lines
513 B
Python
19 lines
513 B
Python
|
|
import multiprocessing
|
|
from partition_main import partition_diversity
|
|
|
|
if __name__ == "__main__":
|
|
cpu_cores = multiprocessing.cpu_count()
|
|
pool = multiprocessing.Pool(processes=cpu_cores)
|
|
|
|
tasks = range(1, 9)
|
|
|
|
image_data_list = pool.map(partition_diversity, tasks)
|
|
|
|
pool.close()
|
|
pool.join()
|
|
|
|
# # 将生成的图片数据保存到本地文件
|
|
# for i, img_data in enumerate(image_data_list, start=1):
|
|
# with open(f'image_{i}.png', 'wb') as f:
|
|
# f.write(img_data) |