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