From 4ea70864fcae1ccb6aee534729e7196379168aba Mon Sep 17 00:00:00 2001 From: Colombo Date: Sun, 23 Feb 2020 20:19:13 +0400 Subject: [PATCH] fix Sorter.py --- mainscripts/Sorter.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mainscripts/Sorter.py b/mainscripts/Sorter.py index 9a8c565..27e95c0 100644 --- a/mainscripts/Sorter.py +++ b/mainscripts/Sorter.py @@ -723,11 +723,11 @@ def sort_by_absdiff(input_path): image_paths = pathex.get_image_paths(input_path) image_paths_len = len(image_paths) - batch_size = 1024 + batch_size = 512 batch_size_remain = image_paths_len % batch_size - i_t = tf.placeholder (tf.float32, (None,256,256,3) ) - j_t = tf.placeholder (tf.float32, (None,256,256,3) ) + i_t = tf.placeholder (tf.float32, (None,None,None,None) ) + j_t = tf.placeholder (tf.float32, (None,None,None,None) ) outputs_full = [] outputs_remain = []