2martens
dbe7b5d844
Added per-class scores to visualise metrics
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-26 12:13:05 +02:00
2martens
6b7c3c9172
Renamed debug module to more fitting 'plotting'
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-24 12:26:43 +02:00
2martens
48ee5466fb
Fixed preparation of predictions for evaluate
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 13:17:31 +02:00
2martens
d24c622d86
Moved GT colour to dark blue for better readability
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 12:46:44 +02:00
2martens
220e0927e0
Moved GT colour to green
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 12:24:58 +02:00
2martens
d16edef94b
Added ability for suffix
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 12:22:12 +02:00
2martens
06be98f59f
Prevented cut off label text
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 12:15:39 +02:00
2martens
34d7128a1c
Made labels more visible
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 12:10:34 +02:00
2martens
5f7870b92f
Fixed check for nesting level of colors array
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 12:10:16 +02:00
2martens
4fc8bb6f8c
Added function to draw single bbox figure
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 11:39:31 +02:00
2martens
554ccc84f9
Moved code into separate functions
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-16 11:04:43 +02:00
2martens
8230488031
Save predictions directly before observations are grouped
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-14 13:43:28 +02:00
2martens
6a990b204a
Fixed per-class grouping of predictions
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 13:48:58 +02:00
2martens
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
2martens
d877da3ef3
Improved readability of code
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 13:00:14 +02:00
2martens
01f43651b5
Fixed ssd train function
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 12:35:52 +02:00
2martens
5bde170138
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 12:25:24 +02:00
2martens
dd2d0f407b
Added config option to enable/disable dropout
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 12:25:13 +02:00
2martens
53022bf1a9
Added saving of predictions before entropy threshold
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 12:15:54 +02:00
2martens
681d23f345
Removed obsolete code
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 11:56:26 +02:00
2martens
64300c3842
Reduced parameter size for predict function
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 11:55:19 +02:00
2martens
90d64ae7c7
Added use_nms config option
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 11:47:18 +02:00
2martens
3d78309603
Made NMS configurable
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-13 11:41:39 +02:00
2martens
343d832545
Added no NMS run to combined graphs
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-12 17:53:15 +02:00
2martens
5e50be22dc
Updated predictions glob string for new suffixes
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-12 17:41:00 +02:00
2martens
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
2martens
c0e7b434dc
Added suffix to save function
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-12 15:23:45 +02:00
2martens
aa114c7237
Added _env to gitignore
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-12 12:06:02 +02:00
2martens
c3286b7169
Set limits to figures
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-11 12:07:51 +02:00
2martens
56a635e5e6
Dealt with different keys in stored metrics files
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-11 11:35:18 +02:00
2martens
9d82b9bd5c
Removed obsolete import
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-11 11:34:55 +02:00
2martens
60b40bc53e
Removed unnecessary argument
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-11 11:29:36 +02:00
2martens
2c226e9975
Added CLI command to visualise a combined graph
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-11 11:25:47 +02:00
2martens
6f543b947e
Removed entropy from structured array
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-09 09:32:48 +02:00
2martens
a91ff8cc22
Fixed filtering for Bayesian SSD
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-09 08:56:08 +02:00
2martens
909d8f7bef
Fixed Bayesian NMS
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-08 22:57:10 +02:00
2martens
ec03b0df4e
Added NMS for Bayesian SSD
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-08 22:00:34 +02:00
2martens
a41f607ca9
Changed filtering to occur per-class
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-08 21:47:00 +02:00
2martens
aaaea5f942
Added if clause to check for empty instance
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-08 21:45:32 +02:00
2martens
eef8839657
Unified naming of evaluation results
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-04 12:12:30 +02:00
2martens
0ae13a2fba
Added support for entropy thresholds in visualise metrics
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-04 11:55:19 +02:00
2martens
de8001c2d3
Save json with indent level 2
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-04 11:41:54 +02:00
2martens
6b56ba9aba
Changed annotation meta file to speed up saving images
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-04 11:39:15 +02:00
2martens
5df06bbfa6
Fixed serialisation of scores
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-03 12:44:35 +02:00
2martens
66d81f3362
Fixed path names
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-03 12:44:21 +02:00
2martens
e1ff56c5d3
Added missing metrics
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-03 12:16:31 +02:00
2martens
f9e489b435
Added attributedict to dependencies
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 16:55:02 +02:00
2martens
32b23bd55d
Overhauled functions to minimize parameter count
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 15:31:28 +02:00
2martens
0498bd03c6
Improved formatting
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:25:22 +02:00
2martens
ee38a74382
Added SonarQube project properties
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:23:33 +02:00
2martens
8aac3ae30f
Added doc comments to public functions in cli module
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:23:10 +02:00
2martens
a8d64ce9c2
Duplicated project dependencies in requirements.txt file
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:20:50 +02:00
2martens
6e85ab86eb
Shortened code for execute action
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:19:00 +02:00
2martens
d58179fa2a
Removed obsolete auto-encoder code
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:10:12 +02:00
2martens
afdf7c96aa
Clarified initialise config function
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 13:00:19 +02:00
2martens
ad96c5b156
Removed obsolete local imports
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 11:02:56 +02:00
2martens
f9c2605603
Simplified evaluate get results function
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-02 10:58:06 +02:00
2martens
686bfcaf40
Fixed evaluation wrt labels and filenames
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-01 13:23:02 +02:00
2martens
9c91f82cc3
Added support for entropy thresholds in evaluate function(s)
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-01 13:05:21 +02:00
2martens
b483fad5af
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-09-01 13:04:45 +02:00
2martens
b49087f415
Added yet another empty check
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 15:43:08 +02:00
2martens
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
2martens
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
2martens
343d7e9bea
Fixed creation of structured array
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 15:00:00 +02:00
2martens
9b3d9a550b
Added top k filtering for bayesian ssd
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 14:02:57 +02:00
2martens
599f7d8957
Fixed indexing
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 13:37:36 +02:00
2martens
cf132a89f0
Fixed application of entropy threshold
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 13:31:13 +02:00
2martens
c403560b99
Added none check
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 12:33:42 +02:00
2martens
f440a03f29
Added checks for empty detection lists
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 12:30:23 +02:00
2martens
ac9d639c10
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 12:30:07 +02:00
2martens
d98434fcfe
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 12:13:40 +02:00
2martens
a09ce1a1b6
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 11:47:54 +02:00
2martens
b4b6c7a777
Implemented Bayesian SSD decoding pipeline
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-29 11:47:39 +02:00
2martens
639d04747d
Provide top_k to decode function
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-12 19:26:12 +02:00
2martens
ef38ba29c1
Changed size of saved images to 640x480
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-12 19:07:34 +02:00
2martens
032b407e42
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-12 18:14:48 +02:00
2martens
9752f285d1
Fixed incompatibilities between numpy arrays and set
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-12 18:11:52 +02:00
2martens
82a370c10f
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-12 18:04:13 +02:00
2martens
d71a135c4b
Optimized retrieving detections in dropout step
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 20:17:52 +02:00
2martens
7be3ee5d4f
Improved formatting
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 18:01:55 +02:00
2martens
30b2cb9b28
Fixed path for pre-trained weights file
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 17:21:59 +02:00
2martens
b645c8d3ff
Added rounding to limit float precision problems
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 16:30:34 +02:00
2martens
9c56c2e514
Skip ground truth boxes that belong to outlier classes
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:58:17 +02:00
2martens
588e66c772
Only the first batch size images are saved on evaluate
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:36:39 +02:00
2martens
6e5cbd3eb3
Fixed the colour space for saving images
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:32:59 +02:00
2martens
39f74fb3bf
Changed data code to load all images for testing
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:30:41 +02:00
2martens
4a8d072b97
Made evaluation compatible with Caffe-like decoding
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:17:33 +02:00
2martens
6ce6e4a195
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:13:53 +02:00
2martens
17fcd4a426
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:12:11 +02:00
2martens
82de20e190
Added support for Caffe implementation to debug module
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 15:03:35 +02:00
2martens
8ad0d50b3b
Changed decoding function to Caffe-like implementation
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 14:56:59 +02:00
2martens
0fbcf0d9ad
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 14:54:16 +02:00
2martens
a223a9dad7
Only load inlier annotations for COCO
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 14:10:06 +02:00
2martens
36a9ca7c83
Improved formatting
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 13:50:34 +02:00
2martens
72c821dc2a
Changed pre-trained weights file to sub-sampled one
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 13:49:46 +02:00
2martens
5a0ae208d3
Added code to sub-sample weights file to 60 classes + background
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-11 13:48:38 +02:00
2martens
facc6c9b1f
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-07 13:45:12 +02:00
2martens
49490aa05b
Colour channels were already swapped
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-07 13:45:02 +02:00
2martens
b123e90821
Updated ssd_keras
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-07 13:24:49 +02:00
2martens
00b756b10a
Added conversion back to RGB before saving images
...
Signed-off-by: Jim Martens <github@2martens.de >
2019-08-07 13:24:01 +02:00