Commit Graph

261 Commits

Author SHA1 Message Date
deeppomf
2815499238
Merge pull request #9 from SoftArmpit/automatic-shape-detection
feat: pass whole images and auto-detect shapes
2018-06-19 15:48:32 -04:00
deeppomf
56ca434ba6
Merge pull request #8 from SoftArmpit/add-dockerfile
build: add dockerfile
2018-06-19 15:46:57 -04:00
SoftArmpit
9847a7f33f feat: pass whole images and auto-detect shapes
We add auto-detection of mask shapes, enabling the ability to put whole
images in the input folder. This is done through OpenCV by checking for
large contours (>150) matching the mask color.

Depends: https://github.com/deeppomf/DeepMindBreak/pull/8
2018-06-15 16:59:22 +01:00
Soft Armpit
c698c124a9 build: add dockerfile
We add a Python 3 Dockerfile. Example usage:

```bash
docker build -t dmb . && docker run dmb
docker cp $(docker ps -alq):/app/decensor_output/ ./
```

In addition we add `requirements.txt` and `.gitignore` files for
convenience.

Closes: https://github.com/deeppomf/DeepMindBreak/issues/6
Closes: https://github.com/deeppomf/DeepMindBreak/issues/7
2018-06-14 18:22:18 +01:00
deeppomf
c38cc156b0 Merge branch 'master' of https://github.com/deeppomf/DeepMindBreak 2018-04-30 17:39:18 -04:00
deeppomf
d5c09954f4 Removed unused folder 2018-04-30 17:39:00 -04:00
deep pomf
8c8204cf45 fix #5 2018-04-11 18:12:28 -04:00
deeppomf
349845a818 update README 2018-03-19 12:21:36 -04:00
deep pomf
0baa81b7c5 clarified requirements 2018-03-15 16:35:33 -04:00
deep pomf
b0e90b4583 updated to reflect new pretrained model 2018-03-15 16:32:56 -04:00
deep pomf
e3e59c4e45 remove stray space 2018-03-15 16:30:03 -04:00
deep pomf
19e2711da7 tweaked format 2018-03-15 16:29:31 -04:00
deep pomf
e231ba7435 fixed license details 2018-03-15 16:00:13 -04:00
deep pomf
359accbb3c added more license details 2018-03-15 15:56:42 -04:00
deeppomf
a66eaff565 added full liscense text 2018-03-14 21:36:17 -04:00
deeppomf
24b87e6150 put correct link and crossed off item on to do list 2018-03-14 21:35:53 -04:00
deeppomf
31ff3e077d updated to new model 2018-03-14 21:34:55 -04:00
deeppomf
a54edbc6c8 updated collage image 2018-03-14 21:17:46 -04:00
deeppomf
85c9886f41 fix typo 2018-03-14 21:04:45 -04:00
deep pomf
1f4f847008 decensor color added to config 2018-03-08 15:11:51 -05:00
deep pomf
028a9b519a change license 2018-03-08 14:45:19 -05:00
deeppomf
1df616ecb3 life happens 2018-03-02 19:54:08 -05:00
deep pomf
f55ad8a087 stop samples from being generated with small datasets 2018-02-28 19:43:55 -05:00
deep pomf
9aeffa0c6c only generate sample images during training when number of test images greater than batch size 2018-02-28 16:10:33 -05:00
deep pomf
a7779d608a Initialize mask_batch and x_batch 2018-02-28 14:49:57 -05:00
deeppomf
018c86c959 added comment about batch_size 2018-02-27 23:15:57 -05:00
deeppomf
3cdf350e18 remove unused folders 2018-02-27 23:11:41 -05:00
deeppomf
faca2f76fd temporarily revert batch_size usage in decensor.py and fix image sample saving in train.py 2018-02-27 23:10:44 -05:00
deep pomf
74e9d08697 fix missing import 2018-02-27 21:32:34 -05:00
deeppomf
4e4c2d74ef use mask_color variable 2018-02-27 11:36:20 -05:00
deeppomf
092c214103 decensor now uses args 2018-02-27 11:34:14 -05:00
deeppomf
967f6d22dd remove folder 2018-02-27 11:28:05 -05:00
deeppomf
971bebec8a added testing output path to args 2018-02-27 11:26:54 -05:00
deeppomf
f9be3bc4bf remove folder 2018-02-26 22:45:54 -05:00
deeppomf
44fb1a32b6 generate training samples folder at runtime 2018-02-26 22:42:59 -05:00
deeppomf
05cbc95ce6 remove print statemetns 2018-02-26 21:48:22 -05:00
deeppomf
57932e1708 fix typo 2018-02-26 21:11:09 -05:00
deeppomf
93e0dc2307 update 2018-02-26 21:07:13 -05:00
deeppomf
6477410f06 remove alpha channel 2018-02-26 21:07:01 -05:00
deeppomf
7e8c8c2967 fix args 2018-02-26 20:36:45 -05:00
deep pomf
76b5903772 update 2018-02-26 10:59:34 -05:00
deep pomf
405456a8f0 more args 2018-02-26 10:59:14 -05:00
deep pomf
c0c817c10d more args 2018-02-26 10:55:48 -05:00
deep pomf
1b28796199 update 2018-02-26 10:47:46 -05:00
deep pomf
9fd51edb95 update 2018-02-26 10:47:10 -05:00
deep pomf
0b008e60df update 2018-02-26 10:43:35 -05:00
deep pomf
1111818451 add config 2018-02-26 10:41:19 -05:00
deep pomf
87e37da0d7 Merge branch 'master' of https://github.com/deeppomf/DeepMindBreak 2018-02-26 10:09:54 -05:00
deeppomf
5b0f34f2e8 train instructions 2018-02-25 09:56:51 -05:00
deeppomf
c0ccc57ea8 better threshold value for mask 2018-02-25 09:47:43 -05:00