 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
|
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 |
|
Jim Martens
|
26904614cb
|
Improved formatting of LICENSE file
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 22:59:17 +01:00 |
|
Jim Martens
|
dddd63aabc
|
Added boilerplate README (needs to be changed)
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 22:57:41 +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 |
|
Jim Martens
|
658973d90d
|
Added weights prefix to parameters
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 15:34:49 +01:00 |
|
Jim Martens
|
88ffd4f879
|
Moved dataset preparation into own function
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 15:32:01 +01:00 |
|
Jim Martens
|
8e0a063759
|
Removed abspath usage in specifying checkpoint dir
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 15:03:47 +01:00 |
|
Jim Martens
|
279e848434
|
Fixed specification of checkpoint dir
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 14:59:50 +01:00 |
|
Jim Martens
|
720ff63c7d
|
Updated saving of checkpoints
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 14:53:45 +01:00 |
|
Jim Martens
|
3c5f332880
|
Added saving of weights after each epoch
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 14:22:27 +01:00 |
|
Jim Martens
|
ec9b64a757
|
Prevented unnecessary early stops
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 13:51:41 +01:00 |
|
Jim Martens
|
948ef64423
|
Added formatting of decoder loss
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 13:41:09 +01:00 |
|
Jim Martens
|
a0094266b3
|
Added training iteration variable to train function
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 13:36:09 +01:00 |
|
Jim Martens
|
3bde913323
|
Extended early stopping with total loss
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 13:27:19 +01:00 |
|
Jim Martens
|
c19ff19820
|
Implemented early stopping
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 13:20:17 +01:00 |
|
Jim Martens
|
682e11b435
|
Added step values to image summaries
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 12:41:10 +01:00 |
|
Jim Martens
|
d84b062bf9
|
Fixed image summaries
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 12:36:36 +01:00 |
|
Jim Martens
|
01db450032
|
Added image summaries
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 12:24:07 +01:00 |
|
Jim Martens
|
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 |
|
Jim Martens
|
fc1aa20396
|
Improved saving location of summaries
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 11:37:35 +01:00 |
|
Jim Martens
|
5ba3bc552f
|
Removed caching of dataset
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 11:37:05 +01:00 |
|
Jim Martens
|
85d743c058
|
Added inlier class to saved weights
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 10:46:25 +01:00 |
|
Jim Martens
|
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 |
|
Jim Martens
|
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 |
|
Jim Martens
|
9d0f93689e
|
Improved paths for saved model weights
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 09:07:33 +01:00 |
|
Jim Martens
|
cc5e7f89a2
|
Removed obsolete shuffling
Signed-off-by: Jim Martens <github@2martens.de>
|
2019-02-08 08:51:43 +01:00 |