From 4f269ad938e1eb3391d0d31c9a8e094fd66fdd55 Mon Sep 17 00:00:00 2001 From: K4YT3X Date: Mon, 5 Nov 2018 15:20:01 -0500 Subject: [PATCH] fixed thread name error --- video2x.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/video2x.py b/video2x.py index 398adc2..892bc5a 100755 --- a/video2x.py +++ b/video2x.py @@ -151,7 +151,7 @@ def upscale_frames(w2): # Create a folder for each thread and append folder # name into a list - thread_folders = [] + thread_pool = [] for thread_id in range(args.threads): thread_folder = '{}\\{}'.format(FRAMES, str(thread_id)) @@ -161,21 +161,21 @@ def upscale_frames(w2): os.mkdir(thread_folder) # Append folder path into list - thread_folders.append(thread_folder) + thread_pool.append((thread_folder, thread_id)) # Evenly distribute images into each folder # until there is none left in the folder for image in frames: # Move image - shutil.move(image, thread_folders[0]) + shutil.move(image, thread_pool[0][0]) # Rotate list - thread_folders = thread_folders[-1:] + thread_folders[:-1] + thread_pool = thread_pool[-1:] + thread_pool[:-1] # Create threads and start them - for thread_folder in thread_folders: + for thread_info in thread_pool: # Create thread - thread = threading.Thread(target=w2.upscale, args=(thread_folder, UPSCALED, args.width, args.height)) - thread.name = str(thread_id) + thread = threading.Thread(target=w2.upscale, args=(thread_info[0], UPSCALED, args.width, args.height)) + thread.name = thread_info[1] # Add threads into the pool upscaler_threads.append(thread)