Commit Graph

570 Commits

Author SHA1 Message Date
ab24c96515 Fixed required packages
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-09 14:07:42 +02:00
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
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
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
a8ef3d000f Added argument to specify the inlying class
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:35:32 +02:00
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
d267512312 Added info about default value to docstring
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:30:32 +02:00
05fa489905 Removed sample generation
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:28:27 +02:00
17537b944b Removed left-over reference to early stopping
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:26:56 +02:00
9a1f91892a Fixes typo
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:26:06 +02:00
1f25122749 Removed reference to early stopping from docstring
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:25:34 +02:00
c83a65571b Corrected docstrings
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:24:05 +02:00
53955cbc46 Removed imports of tensorflow shortcuts
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:22:07 +02:00
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
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
655b3a63d4 Converted to f strings
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 17:00:09 +02:00
df733e69c4 Optimised imports
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 16:57:03 +02:00
84bd2c2e7b Added data loader for COCO data set
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 16:56:50 +02:00
f6699b8444 Added pycocotools as dependency
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-04 14:23:30 +02:00
05ea13d252 Moved COCO wrapper into package
Signed-off-by: Jim Martens <github@2martens.de>
2019-04-01 12:54:58 +02:00
855d8764d0 Removed obsolete log outputs
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-30 12:15:05 +01:00
8ef7d5f999 Removed obsolete parameters
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-25 14:33:41 +01:00
6835c284af Fixed coding style
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-25 14:32:08 +01:00
9a08ea3bb9 Added simple auto-encoder training
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-25 14:31:26 +01:00
0eab8cd123 Added mapping from COCO to WNID
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-25 13:43:54 +01:00
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
ae109fd131 Switched output saving to numpy format
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-21 17:38:22 +01:00
69cc07ed12 Corrected type hinting
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-21 17:31:42 +01:00
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
7ba3333e28 Updated reference to ssd_keras repository
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-21 17:12:35 +01:00
9eb1283b8b Added prediction functionality to ssd module
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-21 17:12:17 +01:00
f40d04e34c Updated reference to new ssd_keras version
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 16:53:04 +01:00
5a7bd6ea9d Fixed image size constant
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 16:52:47 +01:00
b7825c0095 Implemented training function for SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 16:46:14 +01:00
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
9a7a906acf Added ability to load trained weights
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 14:54:28 +01:00
6ac3b5fdcf Added SSD and DropoutSSD classes
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 14:34:46 +01:00
a9d73a1f47 Added empty ssd module
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 13:17:56 +01:00
c090e2c100 Added subpackage info about SSD
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-20 13:15:58 +01:00
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
262a36a68b Added ssd_keras sub-module
Signed-off-by: Jim Martens <github@2martens.de>
2019-03-19 17:21:09 +01:00
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
26904614cb Improved formatting of LICENSE file
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:59:17 +01:00
dddd63aabc Added boilerplate README (needs to be changed)
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:57:41 +01:00
c7a78c35a4 Added package documentation to masterthesis
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:56:58 +01:00
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
9909b5fdfc Improved docstrings in model module
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:42:23 +01:00
1e422d3fa7 Improved imports according to Google standards
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:37:29 +01:00
422e01a917 Improved docstring in aae package
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 22:23:24 +01:00
85f3100b5d Updated docstring for model module
Signed-off-by: Jim Martens <github@2martens.de>
2019-02-08 21:57:57 +01:00