Jim Martens
|
82a370c10f
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-12 18:04:13 +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
|
7be3ee5d4f
|
Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 18:01:55 +02:00 |
Jim Martens
|
30b2cb9b28
|
Fixed path for pre-trained weights file
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 17:21:59 +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
|
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 |
Jim Martens
|
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 |
Jim Martens
|
6e5cbd3eb3
|
Fixed the colour space for saving images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 15:32:59 +02:00 |
Jim Martens
|
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 |
Jim Martens
|
4a8d072b97
|
Made evaluation compatible with Caffe-like decoding
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 15:17:33 +02:00 |
Jim Martens
|
6ce6e4a195
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 15:13:53 +02:00 |
Jim Martens
|
17fcd4a426
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 15:12:11 +02:00 |
Jim Martens
|
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 |
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
|
0fbcf0d9ad
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 14:54:16 +02:00 |
Jim Martens
|
a223a9dad7
|
Only load inlier annotations for COCO
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 14:10:06 +02:00 |
Jim Martens
|
36a9ca7c83
|
Improved formatting
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-11 13:50:34 +02:00 |
Jim Martens
|
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 |
Jim Martens
|
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 |
Jim Martens
|
facc6c9b1f
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-07 13:45:12 +02:00 |
Jim Martens
|
49490aa05b
|
Colour channels were already swapped
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-07 13:45:02 +02:00 |
Jim Martens
|
b123e90821
|
Updated ssd_keras
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-07 13:24:49 +02:00 |
Jim Martens
|
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 |
Jim Martens
|
05d113a4ba
|
Convert image input to BGR
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-07 13:20:34 +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
|
37d5e32031
|
Added confidence threshold to config
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-08-05 13:46:04 +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
|
6576f3d5e1
|
Removed mapping from 'tap' to fire hydrant
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 16:54:06 +02:00 |
Jim Martens
|
6034198262
|
Changed limit for used images to 32
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 15:33:41 +02:00 |
Jim Martens
|
505ea330fc
|
Added missing creation of visualisation output directory
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 15:30:54 +02:00 |
Jim Martens
|
45b5fda185
|
Made visualise work with all tarball ids
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 15:21:49 +02:00 |
Jim Martens
|
8c31794aad
|
Make sure trajectories are sorted before preparing them for Python
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 13:30:01 +02:00 |
Jim Martens
|
4a4c23a66e
|
Discard all images without gt again
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 13:01:34 +02:00 |
Jim Martens
|
e213ed62ec
|
Fixed the number of images used for evaluation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 13:01:04 +02:00 |
Jim Martens
|
60a9d63a1d
|
Changed code to keep images without GT during evaluation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-25 12:04:47 +02:00 |
Jim Martens
|
84adbe21fe
|
Added code to visualise f1 scores to open set error
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-24 12:52:31 +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
|
5cba4046a5
|
Converted category ids to classes for SSD COCO case
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-24 11:59:09 +02:00 |
Jim Martens
|
3330ce69e7
|
Added missing function argument
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-24 11:33:13 +02:00 |
Jim Martens
|
856aad8e21
|
Added code to only take 30 frames of each trajectory in evaluation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 15:39:25 +02:00 |
Jim Martens
|
7f14efcb6a
|
Removed obsolete code
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 14:21:14 +02:00 |
Jim Martens
|
f0a996423b
|
Removed debug print
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 13:57:13 +02:00 |
Jim Martens
|
79c5df1805
|
Added debug print to visualise_metrics
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 13:48:07 +02:00 |
Jim Martens
|
f950504f20
|
Added missing network selection
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 13:46:15 +02:00 |
Jim Martens
|
53472a8342
|
Added code to visualise metrics
For starters, only precision/recall is visualised
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 13:14:22 +02:00 |
Jim Martens
|
d6047665ef
|
Added functionality to use SSD with COCO data set
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-23 11:21:06 +02:00 |
Jim Martens
|
e20167182f
|
Added measurement of usable frames
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-22 16:32:47 +02:00 |
Jim Martens
|
7359f52158
|
Disabled augmentation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-22 16:09:19 +02:00 |
Jim Martens
|
2b656057af
|
Enabled augmentations for training again
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-22 15:20:46 +02:00 |
Jim Martens
|
d1dadabc0d
|
Removed guard clause and use 0 as precision/recall if no class predictions
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-07-18 15:06:34 +02:00 |