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

Improve argmax usage

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-02 16:00:58 +02:00
parent 4595fa85d4
commit d9b742f61a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/cli.py
View File

@ -121,7 +121,7 @@ def _ssd_train(args: argparse.Namespace) -> None:
ymin = instance[-11] * image_size ymin = instance[-11] * image_size
xmax = instance[-10] * image_size xmax = instance[-10] * image_size
ymax = instance[-9] * image_size ymax = instance[-9] * image_size
class_id = np.argmax(instance[:-12]) class_id = np.argmax(instance[:-12], axis=0)
color = colors[class_id] color = colors[class_id]
label = f"{classes_to_names[class_id]}" label = f"{classes_to_names[class_id]}"
current_axis.add_patch( current_axis.add_patch(