Commit Graph
664 Commits
Author SHA1 Message Date
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