 2martens
|
f66bf31785
|
Temporarily disabled save images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-15 10:59:04 +02:00 |
|
|
2martens
|
a521bc4024
|
Moved print statements to predict loop
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-15 10:57:06 +02:00 |
|
|
2martens
|
78e00449ec
|
Fixed print statement
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-15 10:54:04 +02:00 |
|
|
2martens
|
2ecc0023f8
|
Added debug prints to get shapes
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-15 10:46:15 +02:00 |
|
|
2martens
|
c0d23a731b
|
Eliminate side effects from debug function
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-15 10:43:18 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
d77c27727f
|
Fixed transformation of predictions (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 17:34:26 +02:00 |
|
|
2martens
|
ba3cfa379e
|
Fixed transformation of predictions
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 17:33:08 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
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 |
|
|
2martens
|
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 |
|
|
2martens
|
8ef685dae9
|
Fixed unbatching of dict (2nd)
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 17:09:51 +02:00 |
|
|
2martens
|
8d1d78bdb6
|
Fixed unbatching of dict
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 17:09:21 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
50476fc3c0
|
Fixed name of accessed config value
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 16:25:18 +02:00 |
|
|
2martens
|
0d527cf26d
|
Moved general purpose functions up
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 16:23:32 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
ab9f3f8a94
|
Removed obsolete constant N_CLASSES
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 16:06:05 +02:00 |
|
|
2martens
|
af62d0492c
|
Fixed call of save function
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 15:03:39 +02:00 |
|
|
2martens
|
73e4920cc4
|
Fixed mode for ssd model
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 15:01:49 +02:00 |
|
|
2martens
|
fef7a1eac1
|
Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 14:59:47 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
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 |
|
|
2martens
|
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 |
|
|
2martens
|
718467e43c
|
Moved private function to bottom
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:32:37 +02:00 |
|
|
2martens
|
bc8c8b1c6f
|
Removed not needed return doc
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:31:03 +02:00 |
|
|
2martens
|
aa6960ceb2
|
Fixed type hint
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:30:19 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
06548dc8e5
|
Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:29:31 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
ba2bd776ea
|
Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-11 12:22:10 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
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 |
|
|
2martens
|
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 |
|
|
2martens
|
253a507c90
|
Fixed indexing of instance array
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:36:33 +02:00 |
|
|
2martens
|
b5058d9ee9
|
Updated debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:33:15 +02:00 |
|
|
2martens
|
82689ba6ec
|
Added debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:31:02 +02:00 |
|
|
2martens
|
ce87be6019
|
Added conversion to numpy array
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:29:35 +02:00 |
|
|
2martens
|
25de799d80
|
Changed internal coordinates to corners
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:28:46 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
a97b9d8d21
|
Set internal coordinates to "corners"
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 17:12:15 +02:00 |
|
|
2martens
|
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 |
|
|
2martens
|
d517c343be
|
Modified code to save images before and after encoding
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:16:08 +02:00 |
|
|
2martens
|
c2d8da39c8
|
Fixed calculation of bbox coordinates
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:11:14 +02:00 |
|
|
2martens
|
4f6611ec6c
|
Updated train_keras to work with SSD model directly
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:05:31 +02:00 |
|
|
2martens
|
67c098b0d6
|
Added conversion to int for class_id
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 16:01:18 +02:00 |
|
|
2martens
|
aaec64786c
|
Added debug print of class_id
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:55:01 +02:00 |
|
|
2martens
|
7969716e6d
|
Made function compatible with encoded and decoded labels
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:51:20 +02:00 |
|
|
2martens
|
e6370ee92e
|
Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:46:08 +02:00 |
|
|
2martens
|
093cbe472e
|
Changed code to save debug images from processed but not encoded labels
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-10 15:38:06 +02:00 |
|