2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
127 Commits 1 Branch 0 Tags
28ac1a2027b6f9ae569f6b7fda8a6690fde539ab
Commit Graph
115 Commits
Author SHA1 Message Date
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 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 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 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
2martens 5a7bd6ea9d Fixed image size constant 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 16:52:47 +01:00
2martens b7825c0095 Implemented training function for SSD 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 16:46:14 +01:00
2martens 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
2martens 9a7a906acf Added ability to load trained weights 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 14:54:28 +01:00
2martens 6ac3b5fdcf Added SSD and DropoutSSD classes 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 14:34:46 +01:00
2martens a9d73a1f47 Added empty ssd module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 13:17:56 +01:00
2martens c090e2c100 Added subpackage info about SSD 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 13:15:58 +01:00
2martens b9519a516d Added dropout SSD and type hinting for used models 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 13:13:55 +01:00
2martens 262a36a68b Added ssd_keras sub-module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-19 17:21:09 +01:00
2martens c7a78c35a4 Added package documentation to masterthesis 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:56:58 +01:00
2martens 416cb4c7b5 Reformatted code to ensure all parameters are on their own line 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:42:49 +01:00
2martens 9909b5fdfc Improved docstrings in model module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:42:23 +01:00
2martens 1e422d3fa7 Improved imports according to Google standards 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:37:29 +01:00
2martens 422e01a917 Improved docstring in aae package 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:23:24 +01:00