Commit Graph

174 Commits

Author SHA1 Message Date
Jim Martens 8230488031 Save predictions directly before observations are grouped
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-14 13:43:28 +02:00
Jim Martens 75f4e4cb1e Optimised position of saving predictions after observations are grouped
Before the entropy threshold is applied, there is no difference
between the entropy thresholds. Therefore, the predictions should only
be saved once.

Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:05:09 +02:00
Jim Martens d877da3ef3 Improved readability of code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:00:14 +02:00
Jim Martens dd2d0f407b Added config option to enable/disable dropout
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:25:13 +02:00
Jim Martens 53022bf1a9 Added saving of predictions before entropy threshold
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:15:54 +02:00
Jim Martens 681d23f345 Removed obsolete code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:56:26 +02:00
Jim Martens 64300c3842 Reduced parameter size for predict function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:55:19 +02:00
Jim Martens 90d64ae7c7 Added use_nms config option
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:47:18 +02:00
Jim Martens 3d78309603 Made NMS configurable
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:41:39 +02:00
Jim Martens b861270e51 Save predictions directly after entropy threshold is applied as well
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 15:24:33 +02:00
Jim Martens c0e7b434dc Added suffix to save function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 15:23:45 +02:00
Jim Martens 6f543b947e Removed entropy from structured array
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-09 09:32:48 +02:00
Jim Martens a91ff8cc22 Fixed filtering for Bayesian SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-09 08:56:08 +02:00
Jim Martens 909d8f7bef Fixed Bayesian NMS
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 22:57:10 +02:00
Jim Martens ec03b0df4e Added NMS for Bayesian SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 22:00:34 +02:00
Jim Martens a41f607ca9 Changed filtering to occur per-class
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 21:47:00 +02:00
Jim Martens b49087f415 Added yet another empty check
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 15:43:08 +02:00
Jim Martens 25e39edbcd Added check for empty detections in top k application
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 15:04:32 +02:00
Jim Martens da322e287f Made sure top k is only applied when necessary
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 15:00:23 +02:00
Jim Martens 343d7e9bea Fixed creation of structured array
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 15:00:00 +02:00
Jim Martens 9b3d9a550b Added top k filtering for bayesian ssd
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 14:02:57 +02:00
Jim Martens 599f7d8957 Fixed indexing
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 13:37:36 +02:00
Jim Martens cf132a89f0 Fixed application of entropy threshold
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 13:31:13 +02:00
Jim Martens c403560b99 Added none check
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 12:33:42 +02:00
Jim Martens f440a03f29 Added checks for empty detection lists
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 12:30:23 +02:00
Jim Martens b4b6c7a777 Implemented Bayesian SSD decoding pipeline
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-29 11:47:39 +02:00
Jim Martens 639d04747d Provide top_k to decode function
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-12 19:26:12 +02:00
Jim Martens 9752f285d1 Fixed incompatibilities between numpy arrays and set
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-12 18:11:52 +02:00
Jim Martens d71a135c4b Optimized retrieving detections in dropout step
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-11 20:17:52 +02:00
Jim Martens b645c8d3ff Added rounding to limit float precision problems
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-11 16:30:34 +02:00
Jim Martens 8ad0d50b3b Changed decoding function to Caffe-like implementation
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-11 14:56:59 +02:00
Jim Martens 29522921f1 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-05 13:48:21 +02:00
Jim Martens f4982e364c Made iou threshold actually count
Signed-off-by: Jim Martens <github@2martens.de>
2019-08-05 13:44:34 +02:00
Jim Martens 761f6124e0 Prevented overwriting of saved images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-24 12:04:44 +02:00
Jim Martens 84df7b64fa Really fixed entropy threshold keyword argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 14:00:17 +02:00
Jim Martens 3b2f9fe4da Fixed entropy threshold keyword argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-18 13:27:40 +02:00
Jim Martens 28a0d35d36 Implemented support for predicting with range of entropy thresholds
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 12:38:22 +02:00
Jim Martens d9edd94b99 Added entropy threshold as configurable option
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-17 11:54:28 +02:00
Jim Martens 13252b37b2 Changed code so that images are only saved on first batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:59:35 +02:00
Jim Martens 779249afff Added save image call after decoding images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:31:05 +02:00
Jim Martens e0da7eca04 Save images only from one batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:07:20 +02:00
Jim Martens 5c9eaab6e4 Fixed function calls
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:03:09 +02:00
Jim Martens f66bf31785 Temporarily disabled save images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:59:04 +02:00
Jim Martens a521bc4024 Moved print statements to predict loop
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:57:06 +02:00
Jim Martens 78e00449ec Fixed print statement
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:54:04 +02:00
Jim Martens 2ecc0023f8 Added debug prints to get shapes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:46:15 +02:00
Jim Martens 5827d92b8c Added function to save images directly after prediction
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:34:48 +02:00
Jim Martens d77c27727f Fixed transformation of predictions (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:34:26 +02:00
Jim Martens ba3cfa379e Fixed transformation of predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:33:08 +02:00
Jim Martens ab9f3f8a94 Removed obsolete constant N_CLASSES
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:06:05 +02:00