2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed cleaning of labels

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-03 11:32:01 +02:00
parent 45d1a31129
commit 2ba17a1540
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/twomartens/masterthesis/evaluate.py
View File

@ -48,7 +48,6 @@ def get_number_gt_per_class(labels: Sequence[Sequence[Sequence[int]]],
# iterate over images
for i in label_range:
boxes = np.asarray(labels[i])
print(boxes)
# iterate over boxes in image
for j in range(boxes.shape[0]):

8
src/twomartens/masterthesis/main.py
View File

@ -208,10 +208,10 @@ def _ssd_test(args: argparse.Namespace) -> None:
# get labels per batch
_labels = np.asarray(pickle.load(file))
# exclude padded label entries
print(_labels.shape)
cleaned_labels = _labels[_labels[:, :, 0] != -1]
print(cleaned_labels.shape)
labels.extend(cleaned_labels)
for i in range(_labels.shape[0]):
image_labels = _labels[i]
image_labels = image_labels[image_labels[:, 0] != -1]
labels.append(image_labels)
# store labels for later use
with open(label_file, "wb") as file:
pickle.dump(labels, file)