2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
160 Commits 1 Branch 0 Tags
9db610f6a08bbce76d0eebce3e0e46306e441b7f
Commit Graph

146 Commits

Author SHA1 Message Date
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
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
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
Jim Martens
a9d73a1f47 Added empty ssd module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 13:17:56 +01:00
Jim Martens
c090e2c100 Added subpackage info about SSD 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-20 13:15:58 +01:00
Jim Martens
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
Jim Martens
262a36a68b Added ssd_keras sub-module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-03-19 17:21:09 +01:00
Jim Martens
c7a78c35a4 Added package documentation to masterthesis 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:56:58 +01:00
Jim Martens
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
Jim Martens
9909b5fdfc Improved docstrings in model module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:42:23 +01:00
Jim Martens
1e422d3fa7 Improved imports according to Google standards 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:37:29 +01:00
Jim Martens
422e01a917 Improved docstring in aae package 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:23:24 +01:00
Jim Martens
85f3100b5d Updated docstring for model module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:57:57 +01:00
Jim Martens
81f245be99 Updated docstring for util module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:22:00 +01:00
Jim Martens
9bf7229b1e Added TODOs to train module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:14:45 +01:00
Jim Martens
90fa27d9ef Replaced all binary cross entropy losses with log_loss 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:01:38 +01:00
Jim Martens
a51b535728 Improved naming for histogram summaries and added variable summaries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 20:47:34 +01:00
Jim Martens
0cbab44f01 Fixed summaries for histograms 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 20:25:56 +01:00
Jim Martens
a319cf988e Added summaries for grads and learning rate 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 20:13:21 +01:00
Jim Martens
e85b850c63 Fixed frequency of summary savings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 19:41:08 +01:00
Jim Martens
a8f8015ad8 Improved summary logs 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 19:27:33 +01:00
Jim Martens
a2aa794d47 Fixed sample image summary 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:58:19 +01:00
Jim Martens
3b9742a1b4 Improved summary logging 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:40:41 +01:00
Jim Martens
75d7c769c7 Made early stopping conditional and turned it off 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:22:25 +01:00
Jim Martens
ea1b4943cf Fixed function call 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:12:33 +01:00
Jim Martens
d9a35893af Added epoch counter to restored variables 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:12:02 +01:00
Jim Martens
1d4f56e40d Improved docstrings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 17:12:41 +01:00
Jim Martens
028513b404 Renamed internal functions to make them protected 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 16:38:58 +01:00
Jim Martens
483cb4eb3e Removed obsolete function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 16:35:00 +01:00
Jim Martens
2d1fee8048 Moved training of epoch into separate function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 16:34:23 +01:00
Jim Martens
6f36aa7faf Added more parameters to checkpoint 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 15:45:27 +01:00