2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
104 Commits 1 Branch 0 Tags
df733e69c496b6f7338c046b488b2e08808f67f7
Commit Graph
100 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
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
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 be4ee74537 Updated README with badges, links and some text 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 23:40:45 +01:00
2martens 26904614cb Improved formatting of LICENSE file 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:59:17 +01:00
2martens dddd63aabc Added boilerplate README (needs to be changed) 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 22:57:41 +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
2martens 85f3100b5d Updated docstring for model module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:57:57 +01:00
2martens 81f245be99 Updated docstring for util module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:22:00 +01:00
2martens 9bf7229b1e Added TODOs to train module 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 21:14:45 +01:00
2martens 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
2martens 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
2martens 0cbab44f01 Fixed summaries for histograms 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 20:25:56 +01:00
2martens a319cf988e Added summaries for grads and learning rate 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 20:13:21 +01:00
2martens e85b850c63 Fixed frequency of summary savings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 19:41:08 +01:00
2martens a8f8015ad8 Improved summary logs 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 19:27:33 +01:00
2martens a2aa794d47 Fixed sample image summary 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:58:19 +01:00
2martens 3b9742a1b4 Improved summary logging 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:40:41 +01:00
2martens 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
2martens ea1b4943cf Fixed function call 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:12:33 +01:00
2martens d9a35893af Added epoch counter to restored variables 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 18:12:02 +01:00
2martens 1d4f56e40d Improved docstrings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 17:12:41 +01:00
2martens 028513b404 Renamed internal functions to make them protected 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 16:38:58 +01:00
2martens 483cb4eb3e Removed obsolete function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 16:35:00 +01:00
2martens 2d1fee8048 Moved training of epoch into separate function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 16:34:23 +01:00
2martens 6f36aa7faf Added more parameters to checkpoint 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 15:45:27 +01:00
2martens 658973d90d Added weights prefix to parameters 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 15:34:49 +01:00
2martens 88ffd4f879 Moved dataset preparation into own function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 15:32:01 +01:00
2martens 8e0a063759 Removed abspath usage in specifying checkpoint dir 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 15:03:47 +01:00
2martens 279e848434 Fixed specification of checkpoint dir 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 14:59:50 +01:00
2martens 720ff63c7d Updated saving of checkpoints 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 14:53:45 +01:00
2martens 3c5f332880 Added saving of weights after each epoch 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 14:22:27 +01:00
2martens ec9b64a757 Prevented unnecessary early stops 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 13:51:41 +01:00
2martens 948ef64423 Added formatting of decoder loss 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 13:41:09 +01:00
2martens a0094266b3 Added training iteration variable to train function 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 13:36:09 +01:00
2martens 3bde913323 Extended early stopping with total loss 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 13:27:19 +01:00
2martens c19ff19820 Implemented early stopping 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 13:20:17 +01:00
2martens 682e11b435 Added step values to image summaries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 12:41:10 +01:00
2martens d84b062bf9 Fixed image summaries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 12:36:36 +01:00
2martens 01db450032 Added image summaries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 12:24:07 +01:00
2martens 7adb703292 Removed the actual image saving and improved docstrings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 12:23:39 +01:00
2martens fc1aa20396 Improved saving location of summaries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 11:37:35 +01:00
2martens 5ba3bc552f Removed caching of dataset 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 11:37:05 +01:00
2martens 85d743c058 Added inlier class to saved weights 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 10:46:25 +01:00
2martens ce6cda5351 Added comments to give context to the folds 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 10:46:00 +01:00
2martens 3f08c54679 Turned-off implicit summary saving at end of epoch 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 10:32:28 +01:00
2martens 9d0f93689e Improved paths for saved model weights 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 09:07:33 +01:00
2martens cc5e7f89a2 Removed obsolete shuffling 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 08:51:43 +01:00
2martens 2112e1dd0c Added summary writer 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 08:39:59 +01:00
2martens ab52f0feb0 Improved names of loss metrics 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 08:30:06 +01:00
2martens 16e4aaa348 Fixed loss logging 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 08:29:01 +01:00
2martens 758cdb5520 Moved to tf.data.Dataset API 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 08:28:41 +01:00
2martens d9cd24f769 Added verbosity flag 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 07:06:16 +01:00
2martens 1041f24b99 Removed obsolete loss history code 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 07:01:38 +01:00
2martens 4dda663e3b Added resetting the states of the metrics 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 06:59:57 +01:00
2martens a6e90e70b3 Added logging of loss values as summaries 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 06:51:40 +01:00
2martens 65638974d3 Specified correct datatype for global step variable 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 06:48:13 +01:00
2martens 8aee9817f0 Removed obsolete code 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 06:38:56 +01:00
2martens c82f5bf87c Fixed docstrings 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 06:38:27 +01:00
2martens 14cb27afd7 Extracted training steps of models into separate functions 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 06:37:45 +01:00
2martens 887435ddd7 Introduced global steps for each optimizer 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 05:55:19 +01:00
2martens bb76a51d76 Switched to keras binary_crossentropy loss 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-08 05:42:54 +01:00
2martens c52a3cdfb9 Added names and types to loss metrics 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 18:06:33 +01:00
2martens cce0975a13 Realized learning rate decay with variable 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 17:55:51 +01:00
2martens ebfa67abb0 Fixed saving loss history 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 17:55:24 +01:00
2martens abf52a88b0 Properly created global step 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 17:29:37 +01:00
2martens 661e99c9f5 Added required __init__ file 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 16:58:50 +01:00
2martens bfc888389f Added missing dependencies and switched to GPU dependency 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 16:58:32 +01:00
2martens ecbfbf2a2a Added mnist training function for eager mode 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 16:57:45 +01:00
2martens d4dc705dfb Added utility functions to save prediction results as images 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 16:21:07 +01:00
2martens ca31ae2fbc Added mnist compatible models for AAE 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-07 15:58:24 +01:00
2martens cf43f81053 Exchanged keras for tensorflow 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-05 15:52:35 +01:00
2martens 3396d00e0a [#1] Added setup.py file 
Closes #1

Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-05 13:37:47 +01:00
2martens 2caa6bc8f9 Added empty python packages 
Signed-off-by: Jim Martens <github@2martens.de>
 2019-02-04 16:52:42 +01:00