diff --git a/jfda/minibatch.py b/jfda/minibatch.py index 23eda10..75f7b91 100644 --- a/jfda/minibatch.py +++ b/jfda/minibatch.py @@ -8,7 +8,7 @@ class MiniBatcher(multiprocessing.Process): '''generate minibatch - given a queue, put (positives, negatives, part faces, landmark faces) = (n1, n2, n3, n4) + given a queue, put (negatives, positives, part faces, landmark faces) = (n1, n2, n3, n4) ''' def __init__(self, db_names, ns, net_type): diff --git a/layers/data_layer.py b/layers/data_layer.py index 8186365..24be6af 100644 --- a/layers/data_layer.py +++ b/layers/data_layer.py @@ -40,7 +40,7 @@ def set_batch_num(self, n1, n2, n3, n4): self.net_input_size = cfg.NET_INPUT_SIZE[cfg.NET_TYPE] def set_data_queue(self, queue): - '''the queue should put a minibatch with size of (positives, negatives, part faces, landmark faces) = + '''the queue should put a minibatch with size of (negatives, positives, part faces, landmark faces) = (n1, n2, n3, n4) in a dict ''' self.data_queue = queue