2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity
127 Commits 1 Branch 0 Tags
28ac1a2027b6f9ae569f6b7fda8a6690fde539ab
Commit Graph
100 Commits
This Branch
This Branch
All Branches
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 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
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 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