Commit Graph

625 Commits

Author SHA1 Message Date
5aac93f444 Modified mean average precision to only use classes with ground truth
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:44:52 +02:00
df2faaffe9 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:30:12 +02:00
3ec6f9ad0a Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 12:18:09 +02:00
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
135144efc8 Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:46:41 +02:00
779249afff Added save image call after decoding images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:31:05 +02:00
e0da7eca04 Save images only from one batch
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:07:20 +02:00
5c9eaab6e4 Fixed function calls
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 11:03:09 +02:00
f66bf31785 Temporarily disabled save images
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:59:04 +02:00
a521bc4024 Moved print statements to predict loop
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:57:06 +02:00
78e00449ec Fixed print statement
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:54:04 +02:00
2ecc0023f8 Added debug prints to get shapes
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:46:15 +02:00
c0d23a731b Eliminate side effects from debug function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-15 10:43:18 +02:00
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
d77c27727f Fixed transformation of predictions (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:34:26 +02:00
ba3cfa379e Fixed transformation of predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:33:08 +02:00
38229e1848 Only save images from first batch in evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:19:56 +02:00
c2be08e0c8 Changed code to save images with predictions not ground truth
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:13:17 +02:00
6f6b4b7e34 Fixed debug saving of images to work with predictions
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:12:54 +02:00
8ef685dae9 Fixed unbatching of dict (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:09:51 +02:00
8d1d78bdb6 Fixed unbatching of dict
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:09:21 +02:00
1474f30744 Added code to save images during ssd_evaluate
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 17:08:00 +02:00
50476fc3c0 Fixed name of accessed config value
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:25:18 +02:00
0d527cf26d Moved general purpose functions up
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:23:32 +02:00
d53a0cb6b4 Made ssd_evaluate conform to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:23:01 +02:00
ab9f3f8a94 Removed obsolete constant N_CLASSES
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 16:06:05 +02:00
af62d0492c Fixed call of save function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:03:39 +02:00
73e4920cc4 Fixed mode for ssd model
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 15:01:49 +02:00
fef7a1eac1 Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:59:47 +02:00
e8a82fbfaf Updated ssd_train function to use new signature of predict function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:56:42 +02:00
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
fbcf2c261c Changed predict function to conform to clean code standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 14:52:59 +02:00
718467e43c Moved private function to bottom
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:32:37 +02:00
bc8c8b1c6f Removed not needed return doc
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:31:03 +02:00
aa6960ceb2 Fixed type hint
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:30:19 +02:00
2d19ccf4a5 Changed ssd_train to use new compile_model function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:30:08 +02:00
06548dc8e5 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:29:31 +02:00
b30b332c7c Made ssd_test conform to clean code principles
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:29:06 +02:00
ba2bd776ea Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:22:10 +02:00
0ae33130ad Made image size parameter more compatible with clean code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:22:01 +02:00
9fa525ebe3 Added extra functions to compile model and get loss function
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 12:20:34 +02:00
a848c12d64 Cleaned up ssd module and removed no longer needed code
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-11 11:24:06 +02:00
253a507c90 Fixed indexing of instance array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:36:33 +02:00
b5058d9ee9 Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:33:15 +02:00
82689ba6ec Added debug print
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:31:02 +02:00
ce87be6019 Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:29:35 +02:00
25de799d80 Changed internal coordinates to corners
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:28:46 +02:00
a685bba4fa Properly reversed encoding of labels to visualize them
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:26:16 +02:00
a97b9d8d21 Set internal coordinates to "corners"
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 17:12:15 +02:00
4c4c05d383 Made it possible to save images without preventing training
Signed-off-by: Jim Martens <github@2martens.de>
2019-07-10 16:29:39 +02:00