[P] Deep Danbooru (girl image tag estimation) source code is available on github now
Previous post: https://www.reddit.com/r/MachineLearning/comments/akbc11/p_tag_estimation_for_animestyle_girl_image/
Hi, as promised, I released my code to github now.
Here is brief changes from last post:
- ML library is changed from CNTK to Tensorflow 2.x
- Optimizer is changed from Adam to SGD with momentum
- Dataset is updated (20191108)
Repository URL is here : https://github.com/KichangKim/DeepDanbooru
It contains pre-trained model as well, so you can test your image immediately.
Legacy CNTK version web demo is still available, but URL is changed to http://kanotype.iptime.org:8004/deepdanbooru/
(only port). Current web demo uses latest Tensorflow 2 version. I’ll keep CNTK version demo for a while, but I’ll end it one day.
Have fun.
submitted by /u/KichangKim
[link] [comments]