Commit 84ae5441 authored by Tiago Pereira's avatar Tiago Pereira
Browse files

More optimizations #28

parent 190863f5
Pipeline #11396 canceled with stages
......@@ -89,6 +89,7 @@ class Siamese(Base):
# Finding the right pair
j = i
# TODO: Lame solution. Fix this
while True:
if left_possible_indexes[i] != right_possible_indexes[j]:
class_index.append(right_possible_indexes[j])
......
......@@ -83,7 +83,6 @@ class SiameseMemory(Siamese, Memory):
**Return**
"""
#genuine = True
pairs_generator = self.get_genuine_or_not(self.data, self.labels)
for i in range(self.data.shape[0]):
......@@ -101,5 +100,3 @@ class SiameseMemory(Siamese, Memory):
right = self.normalize_sample(right)
yield left.astype(self.input_dtype), right.astype(self.input_dtype), label
#return [sample_l.astype(self.input_dtype), sample_r.astype(self.input_dtype), labels_siamese]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment