2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
383 Commits 1 Branch 0 Tags
e908248f894e6dfb6a484584665a7ab48e487b8a
Commit Graph

383 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Jim Martens
37b0b702a3 Changed batch size to 16 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-12 13:42:48 +02:00
Jim Martens
1733fc8c99 Added data directory to git ignore 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-12 13:39:20 +02:00
Jim Martens
30d2e156fe Reduced zsize and increases learning rate 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-11 15:20:48 +02:00
Jim Martens
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
Jim Martens
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
Jim Martens
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
Jim Martens
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
Jim Martens
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
Jim Martens
b173746784 Ensured summaries are written 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-10 15:19:44 +02:00
Jim Martens
a7f9d586f4 Fixed loop through data set 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-10 15:18:39 +02:00
Jim Martens
da6e348edc Fixed data loading 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-10 15:17:24 +02:00
Jim Martens
f81ecbac0d Fixed slicing 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-10 15:16:13 +02:00
Jim Martens
f30edeb3f2 Added command line interface 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-09 14:09:22 +02:00
Jim Martens
ab24c96515 Fixed required packages 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-09 14:07:42 +02:00
Jim Martens
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
Jim Martens
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
Jim Martens
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
Jim Martens
a8ef3d000f Added argument to specify the inlying class 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:35:32 +02:00
Jim Martens
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
Jim Martens
d267512312 Added info about default value to docstring 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:30:32 +02:00
Jim Martens
05fa489905 Removed sample generation 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:28:27 +02:00
Jim Martens
17537b944b Removed left-over reference to early stopping 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:26:56 +02:00
Jim Martens
9a1f91892a Fixes typo 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:26:06 +02:00
Jim Martens
1f25122749 Removed reference to early stopping from docstring 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:25:34 +02:00
Jim Martens
c83a65571b Corrected docstrings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:24:05 +02:00
Jim Martens
53955cbc46 Removed imports of tensorflow shortcuts 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:22:07 +02:00
Jim Martens
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
Jim Martens
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
Jim Martens
655b3a63d4 Converted to f strings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 17:00:09 +02:00
Jim Martens
df733e69c4 Optimised imports 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 16:57:03 +02:00
Jim Martens
84bd2c2e7b Added data loader for COCO data set 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 16:56:50 +02:00
Jim Martens
f6699b8444 Added pycocotools as dependency 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-04 14:23:30 +02:00
Jim Martens
05ea13d252 Moved COCO wrapper into package 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-04-01 12:54:58 +02:00
Jim Martens
855d8764d0 Removed obsolete log outputs 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-30 12:15:05 +01:00
Jim Martens
8ef7d5f999 Removed obsolete parameters 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-25 14:33:41 +01:00
Jim Martens
6835c284af Fixed coding style 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-25 14:32:08 +01:00
Jim Martens
9a08ea3bb9 Added simple auto-encoder training 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-25 14:31:26 +01:00
Jim Martens
0eab8cd123 Added mapping from COCO to WNID 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-25 13:43:54 +01:00
Jim Martens
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
Jim Martens
ae109fd131 Switched output saving to numpy format 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-21 17:38:22 +01:00
Jim Martens
69cc07ed12 Corrected type hinting 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-21 17:31:42 +01:00
Jim Martens
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
Jim Martens
7ba3333e28 Updated reference to ssd_keras repository 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-21 17:12:35 +01:00
Jim Martens
9eb1283b8b Added prediction functionality to ssd module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-21 17:12:17 +01:00
Jim Martens
f40d04e34c Updated reference to new ssd_keras version 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 16:53:04 +01:00
Jim Martens
5a7bd6ea9d Fixed image size constant 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 16:52:47 +01:00
Jim Martens
b7825c0095 Implemented training function for SSD 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 16:46:14 +01:00
Jim Martens
167899b2f8 Added iou threshold and top_k parameters to function calls 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 14:55:32 +01:00
Jim Martens
9a7a906acf Added ability to load trained weights 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 14:54:28 +01:00
Jim Martens
6ac3b5fdcf Added SSD and DropoutSSD classes 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 14:34:46 +01:00