2martens
|
d7ef9513a7
|
Fixed comparison of coordinates
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-28 18:19:28 +02:00 |
|
2martens
|
ad11a5f6c6
|
Fixed weights path given to SSD predict
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-28 17:55:41 +02:00 |
|
2martens
|
aa42d831ed
|
Added SSD validation command
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-28 17:53:44 +02:00 |
|
2martens
|
9d611496d9
|
Implemented data retrieval for SceneNet validation data
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-28 17:13:19 +02:00 |
|
2martens
|
c7c948295e
|
Updated referenced ssd_keras version
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-27 10:01:57 +02:00 |
|
2martens
|
93c2d88dd4
|
Remove data that cannot be pickled
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-20 11:19:37 +02:00 |
|
2martens
|
4ab5ba7745
|
Cleaned up imports
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-20 11:18:57 +02:00 |
|
2martens
|
e98196ec0d
|
Fixed data preparation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-19 15:55:48 +02:00 |
|
2martens
|
70ff4b2591
|
Fixed instance file name
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-19 15:20:44 +02:00 |
|
2martens
|
517bc51118
|
Attempted to fix image loading
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-19 15:18:10 +02:00 |
|
2martens
|
5f9e8d52f3
|
Fixed wrong attribute usage
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-19 15:09:46 +02:00 |
|
2martens
|
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 |
|
2martens
|
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 |
|
2martens
|
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 |
|
2martens
|
ae87f08d6c
|
Added WNID to COCO mapping
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-19 10:58:39 +02:00 |
|
2martens
|
b3743fb18b
|
Added scenenet protobuf file
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-18 17:56:50 +02:00 |
|
2martens
|
2140b13857
|
Fixed dimensions of input transform
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-18 12:17:01 +02:00 |
|
2martens
|
71623db905
|
Added dense layer as latent space
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-18 12:13:21 +02:00 |
|
2martens
|
532f4ac567
|
Make bottleneck even harder
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 14:26:30 +02:00 |
|
2martens
|
7249d569f7
|
Fixed zsize for val
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 14:23:51 +02:00 |
|
2martens
|
f4f9d31a7d
|
Reduced latent space to 32 channels
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 14:14:48 +02:00 |
|
2martens
|
600cb55e2e
|
Removed visualization of latent space
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 14:14:27 +02:00 |
|
2martens
|
bc07fda615
|
Simplified use of debug
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 14:04:23 +02:00 |
|
2martens
|
be40a52dde
|
Fixed nested arguments
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 13:51:48 +02:00 |
|
2martens
|
49115ccc1d
|
Added missing argument in function call
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 13:49:20 +02:00 |
|
2martens
|
2e388c0f82
|
Fixed source of validation images (again)
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 13:48:12 +02:00 |
|
2martens
|
0091027e1d
|
Fixed source of validation images
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 13:46:56 +02:00 |
|
2martens
|
851bbb1f43
|
Fixed typo
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 13:27:51 +02:00 |
|
2martens
|
9db610f6a0
|
Renamed use to val
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 13:00:09 +02:00 |
|
2martens
|
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 |
|
2martens
|
d3e212ab32
|
Added debug flag
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 12:55:51 +02:00 |
|
2martens
|
80d4861a4b
|
Added code to load COCO validation data
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 12:46:23 +02:00 |
|
2martens
|
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 |
|
2martens
|
710c8359b3
|
Reduced kernel size further
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 12:06:24 +02:00 |
|
2martens
|
eae908c617
|
Added visualization for latent space
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-17 11:10:59 +02:00 |
|
2martens
|
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 |
|
2martens
|
18fc46b70c
|
Fix model to actually introduce a bottleneck
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-16 14:20:23 +02:00 |
|
2martens
|
7edf5879e8
|
Added run functionality
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-16 11:09:27 +02:00 |
|
2martens
|
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 |
|
2martens
|
823cfdbfcf
|
Removed cython from requirements.txt
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-16 10:26:40 +02:00 |
|
2martens
|
d86c53c18e
|
Save reconstruction image summaries only with verbose on
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-15 17:10:48 +02:00 |
|
2martens
|
2e076597ef
|
Fixed wrong reference
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-15 12:16:38 +02:00 |
|
2martens
|
031a1f9ef0
|
Removed experimental AUTOTUNE
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-15 12:14:06 +02:00 |
|
2martens
|
d6e3cba225
|
Piped verbose option through to train
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-15 12:13:37 +02:00 |
|
2martens
|
1243cc0a4b
|
Removed duplicate loss log
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-15 10:24:16 +02:00 |
|
2martens
|
3afb74587b
|
Updated image paths to work with cvpc7 installation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-15 10:21:36 +02:00 |
|
2martens
|
c8e49bd0da
|
Removed first deconv layer to achieve symmetry in convolutions
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-14 09:36:10 +02:00 |
|
2martens
|
1764e10da4
|
Removed unnecessary layers and fixed the names of the remaining ones
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 14:26:43 +02:00 |
|
2martens
|
32aecaea36
|
Deactivated pooling and upsampling layers
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 14:21:47 +02:00 |
|
2martens
|
1ea9ff98d4
|
Increased zsize to 64
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 14:17:17 +02:00 |
|
2martens
|
24e8d0e009
|
Deactivated dropout
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 14:17:05 +02:00 |
|
2martens
|
d749678d20
|
Removed unnecessary layers
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 14:16:48 +02:00 |
|
2martens
|
568adeab08
|
Implemented auto-encoder model from OliverEdholm
Source: https://github.com/OliverEdholm/Convolutional-Autoencoder/
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 13:45:27 +02:00 |
|
2martens
|
7ea63a9a49
|
Reverted to default AdamOptimizer beta values
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 13:44:05 +02:00 |
|
2martens
|
25e2110a5b
|
Changed size of images to 256, 256
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 13:43:29 +02:00 |
|
2martens
|
37b0b702a3
|
Changed batch size to 16
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 13:42:48 +02:00 |
|
2martens
|
1733fc8c99
|
Added data directory to git ignore
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-12 13:39:20 +02:00 |
|
2martens
|
30d2e156fe
|
Reduced zsize and increases learning rate
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-11 15:20:48 +02:00 |
|
2martens
|
e7be92550d
|
Resized images to 64x64 to alleviate memory problems
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-11 15:20:28 +02:00 |
|
2martens
|
41fe940d5b
|
Modified model as per instructions from Keras blog
https://blog.keras.io/building-autoencoders-in-keras.html
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-11 15:19:50 +02:00 |
|
2martens
|
0d0ea882dd
|
Made filter sizes in encoder/decoder models dependent on variable
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-11 13:31:06 +02:00 |
|
2martens
|
28ac1a2027
|
Fixed mapping from image to bounding box
There are multiple bounding boxes per image. Previously bounding boxes
were wrongly matched.
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-10 17:42:59 +02:00 |
|
2martens
|
a0f659cea2
|
Cleaned up data set to prevent awkward cases
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-10 17:00:32 +02:00 |
|
2martens
|
b173746784
|
Ensured summaries are written
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-10 15:19:44 +02:00 |
|
2martens
|
a7f9d586f4
|
Fixed loop through data set
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-10 15:18:39 +02:00 |
|
2martens
|
da6e348edc
|
Fixed data loading
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-10 15:17:24 +02:00 |
|
2martens
|
f81ecbac0d
|
Fixed slicing
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-10 15:16:13 +02:00 |
|
2martens
|
f30edeb3f2
|
Added command line interface
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-09 14:09:22 +02:00 |
|
2martens
|
ab24c96515
|
Fixed required packages
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-09 14:07:42 +02:00 |
|
2martens
|
523a9c8e71
|
Removed hard-coded reliance on specific batch size
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-09 10:33:14 +02:00 |
|
2martens
|
be501d606f
|
Modified data module to cut everything but the bounding box
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-09 10:18:50 +02:00 |
|
2martens
|
57da44c6d8
|
Modified data loader to supply the full trainval35k data
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 19:04:19 +02:00 |
|
2martens
|
a8ef3d000f
|
Added argument to specify the inlying class
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:35:32 +02:00 |
|
2martens
|
d499c55ab4
|
Added skeleton function for loading SceneNet RGB-D data
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:30:52 +02:00 |
|
2martens
|
d267512312
|
Added info about default value to docstring
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:30:32 +02:00 |
|
2martens
|
05fa489905
|
Removed sample generation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:28:27 +02:00 |
|
2martens
|
17537b944b
|
Removed left-over reference to early stopping
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:26:56 +02:00 |
|
2martens
|
9a1f91892a
|
Fixes typo
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:26:06 +02:00 |
|
2martens
|
1f25122749
|
Removed reference to early stopping from docstring
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:25:34 +02:00 |
|
2martens
|
c83a65571b
|
Corrected docstrings
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:24:05 +02:00 |
|
2martens
|
53955cbc46
|
Removed imports of tensorflow shortcuts
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:22:07 +02:00 |
|
2martens
|
08f26fa9e0
|
Extracted preparation of MNIST data into separate file
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:21:29 +02:00 |
|
2martens
|
403577da0a
|
Extracted adversarial auto-encoder functions into separate file
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:14:11 +02:00 |
|
2martens
|
655b3a63d4
|
Converted to f strings
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 17:00:09 +02:00 |
|
2martens
|
df733e69c4
|
Optimised imports
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 16:57:03 +02:00 |
|
2martens
|
84bd2c2e7b
|
Added data loader for COCO data set
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 16:56:50 +02:00 |
|
2martens
|
f6699b8444
|
Added pycocotools as dependency
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-04 14:23:30 +02:00 |
|
2martens
|
05ea13d252
|
Moved COCO wrapper into package
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-04-01 12:54:58 +02:00 |
|
2martens
|
855d8764d0
|
Removed obsolete log outputs
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-30 12:15:05 +01:00 |
|
2martens
|
8ef7d5f999
|
Removed obsolete parameters
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-25 14:33:41 +01:00 |
|
2martens
|
6835c284af
|
Fixed coding style
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-25 14:32:08 +01:00 |
|
2martens
|
9a08ea3bb9
|
Added simple auto-encoder training
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-25 14:31:26 +01:00 |
|
2martens
|
0eab8cd123
|
Added mapping from COCO to WNID
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-25 13:43:54 +01:00 |
|
2martens
|
db0d275bdd
|
Save predictions batch-wise to prevent memory issues
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-21 17:49:34 +01:00 |
|
2martens
|
ae109fd131
|
Switched output saving to numpy format
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-21 17:38:22 +01:00 |
|
2martens
|
69cc07ed12
|
Corrected type hinting
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-21 17:31:42 +01:00 |
|
2martens
|
976563526a
|
Added train and predict functions to module documentation
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-21 17:14:44 +01:00 |
|
2martens
|
7ba3333e28
|
Updated reference to ssd_keras repository
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-21 17:12:35 +01:00 |
|
2martens
|
9eb1283b8b
|
Added prediction functionality to ssd module
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-21 17:12:17 +01:00 |
|
2martens
|
f40d04e34c
|
Updated reference to new ssd_keras version
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-03-20 16:53:04 +01:00 |
|