Commit Graph

744 Commits

Author SHA1 Message Date
f901943668 Added python 3.8 classifier 2020-03-28 11:07:42 +01:00
a4c195ff29 Added pyproject file 2020-03-28 10:59:24 +01:00
b57d307c2a Fixed remaining typo in README
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 18:31:19 +02:00
029a2e2bb7 Moved templates to correct location
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 18:27:56 +02:00
356a93dee4 Added issue templates
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 18:24:02 +02:00
2c3a1e7f55 Fixed typo wrt master thesis
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:38:49 +02:00
db719416be Added reference to Pierluigi Ferrari's ssd implementation
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:23:24 +02:00
0eb1fee649 Completed documentation
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:19:32 +02:00
abd702411d Added coding info to scenenet protobuf file
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:06:11 +02:00
ae5170bb72 Updated README for PyPi release
Signed-off-by: Jim Martens <github@2martens.de>
2019-10-01 17:03:55 +02:00
8f78426a39 Made visualise metrics compatible with older result files
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-26 12:26:11 +02:00
dbe7b5d844 Added per-class scores to visualise metrics
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-26 12:13:05 +02:00
6b7c3c9172 Renamed debug module to more fitting 'plotting'
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-24 12:26:43 +02:00
48ee5466fb Fixed preparation of predictions for evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 13:17:31 +02:00
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
220e0927e0 Moved GT colour to green
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:24:58 +02:00
d16edef94b Added ability for suffix
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:22:12 +02:00
06be98f59f Prevented cut off label text
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:15:39 +02:00
34d7128a1c Made labels more visible
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 12:10:34 +02:00
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
4fc8bb6f8c Added function to draw single bbox figure
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 11:39:31 +02:00
554ccc84f9 Moved code into separate functions
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-16 11:04:43 +02:00
8230488031 Save predictions directly before observations are grouped
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-14 13:43:28 +02:00
6a990b204a Fixed per-class grouping of predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:48:58 +02:00
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
d877da3ef3 Improved readability of code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 13:00:14 +02:00
01f43651b5 Fixed ssd train function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:35:52 +02:00
5bde170138 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:25:24 +02:00
dd2d0f407b Added config option to enable/disable dropout
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:25:13 +02:00
53022bf1a9 Added saving of predictions before entropy threshold
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 12:15:54 +02:00
681d23f345 Removed obsolete code
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:56:26 +02:00
64300c3842 Reduced parameter size for predict function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:55:19 +02:00
90d64ae7c7 Added use_nms config option
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:47:18 +02:00
3d78309603 Made NMS configurable
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-13 11:41:39 +02:00
343d832545 Added no NMS run to combined graphs
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 17:53:15 +02:00
5e50be22dc Updated predictions glob string for new suffixes
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 17:41:00 +02:00
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
c0e7b434dc Added suffix to save function
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 15:23:45 +02:00
aa114c7237 Added _env to gitignore
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-12 12:06:02 +02:00
c3286b7169 Set limits to figures
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 12:07:51 +02:00
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
9d82b9bd5c Removed obsolete import
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 11:34:55 +02:00
60b40bc53e Removed unnecessary argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-11 11:29:36 +02:00
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
6f543b947e Removed entropy from structured array
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-09 09:32:48 +02:00
a91ff8cc22 Fixed filtering for Bayesian SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-09 08:56:08 +02:00
909d8f7bef Fixed Bayesian NMS
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 22:57:10 +02:00
ec03b0df4e Added NMS for Bayesian SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 22:00:34 +02:00
a41f607ca9 Changed filtering to occur per-class
Signed-off-by: Jim Martens <github@2martens.de>
2019-09-08 21:47:00 +02:00
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