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

Added support for Caffe implementation to debug module

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-08-11 15:03:35 +02:00
parent 8ad0d50b3b
commit 82de20e190
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/twomartens/masterthesis/debug.py
View File

@ -87,6 +87,12 @@ def save_ssd_train_images(images: Union[np.ndarray, Sequence[str]], labels: np.n
ymin = instance[4] ymin = instance[4]
xmax = instance[5] xmax = instance[5]
ymax = instance[6] ymax = instance[6]
elif len(instance) == 6: # predictions using Caffe method
class_id = int(instance[0])
xmin = instance[2]
ymin = instance[3]
xmax = instance[4]
ymax = instance[5]
else: else:
instance = np.copy(instance) instance = np.copy(instance)
class_id = np.argmax(instance[:-12], axis=0) class_id = np.argmax(instance[:-12], axis=0)