Commit Graph

649 Commits

Author SHA1 Message Date
dbab9e36b5 Fixed function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 19:11:19 +02:00
d60979e866 Fixed function name for resizing images
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 19:10:09 +02:00
9b508477f3 Simplified data set creation by removing not required labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 19:06:44 +02:00
7a89a3cf7b Fixed creation of final labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:50:38 +02:00
d12e7ec30d Changed code to get shape of labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:47:51 +02:00
65013a843b Fixed numpy array creation
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:46:21 +02:00
f5db36bf9f Attempt conversion to Numpy array before converting to Tensor
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:44:25 +02:00
497e8c911e Converted tuple to list for list of labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:40:25 +02:00
ee74480dd7 Optimized check for empty labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:38:27 +02:00
2276d359d6 Added check to prevent frames without labels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:34:28 +02:00
5fd8ba1ae8 Fixed background being saved as instance
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:25:29 +02:00
d7ef9513a7 Fixed comparison of coordinates
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 18:19:28 +02:00
ad11a5f6c6 Fixed weights path given to SSD predict
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 17:55:41 +02:00
aa42d831ed Added SSD validation command
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 17:53:44 +02:00
9d611496d9 Implemented data retrieval for SceneNet validation data
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-28 17:13:19 +02:00
c7c948295e Updated referenced ssd_keras version
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-27 10:01:57 +02:00
93c2d88dd4 Remove data that cannot be pickled
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-20 11:19:37 +02:00
4ab5ba7745 Cleaned up imports
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-20 11:18:57 +02:00
e98196ec0d Fixed data preparation
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 15:55:48 +02:00
70ff4b2591 Fixed instance file name
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 15:20:44 +02:00
517bc51118 Attempted to fix image loading
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 15:18:10 +02:00
5f9e8d52f3 Fixed wrong attribute usage
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 15:09:46 +02:00
37e355e820 Implemented ground truth preparation for SceneNet RGB-D
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 15:02:07 +02:00
3704f6a2ce Fixed COCO to WNID and WNID to COCO mappings
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 13:56:13 +02:00
5ab0a28bfa Removed person class from COCO to WNID mapping
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 11:20:30 +02:00
ae87f08d6c Added WNID to COCO mapping
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-19 10:58:39 +02:00
b3743fb18b Added scenenet protobuf file
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-18 17:56:50 +02:00
2140b13857 Fixed dimensions of input transform
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-18 12:17:01 +02:00
71623db905 Added dense layer as latent space
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-18 12:13:21 +02:00
532f4ac567 Make bottleneck even harder
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 14:26:30 +02:00
7249d569f7 Fixed zsize for val
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 14:23:51 +02:00
f4f9d31a7d Reduced latent space to 32 channels
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 14:14:48 +02:00
600cb55e2e Removed visualization of latent space
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 14:14:27 +02:00
bc07fda615 Simplified use of debug
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 14:04:23 +02:00
be40a52dde Fixed nested arguments
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 13:51:48 +02:00
49115ccc1d Added missing argument in function call
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 13:49:20 +02:00
2e388c0f82 Fixed source of validation images (again)
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 13:48:12 +02:00
0091027e1d Fixed source of validation images
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 13:46:56 +02:00
851bbb1f43 Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 13:27:51 +02:00
9db610f6a0 Renamed use to val
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 13:00:09 +02:00
e22de150b0 Corrected default values for train and run functions
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 12:57:26 +02:00
d3e212ab32 Added debug flag
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 12:55:51 +02:00
80d4861a4b Added code to load COCO validation data
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 12:46:23 +02:00
97b1b8963f Added latent space visualization to run situation as well
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 12:08:59 +02:00
710c8359b3 Reduced kernel size further
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 12:06:24 +02:00
eae908c617 Added visualization for latent space
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-17 11:10:59 +02:00
e7e8e15d7f Reduce kernel size for encoder/decoder to 5
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-16 22:08:27 +02:00
18fc46b70c Fix model to actually introduce a bottleneck
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-16 14:20:23 +02:00
7edf5879e8 Added run functionality
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-16 11:09:27 +02:00
109fda6292 Added CLI functionality to use a trained auto-encoder
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-16 11:06:27 +02:00