2martens
/
masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
719 Commits 1 Branch 0 Tags 776 KiB
Commit Graph

172 Commits

Author SHA1 Message Date
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
Jim Martens af62d0492c Fixed call of save function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-07-11 15:03:39 +02:00
Jim Martens 6c431e5e70 Moved public train function before the private functions 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-07-11 14:55:20 +02:00