From 785ebc0ef5e0bfebb7f4109d803676f6c62cbc5b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9B=B9=E4=BD=B3=E9=91=AB?= Date: Mon, 11 Dec 2017 14:03:36 +0800 Subject: [PATCH] Create readme.md --- readme.md | 117 ++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 117 insertions(+) create mode 100644 readme.md diff --git a/readme.md b/readme.md new file mode 100644 index 0000000..fcfd3c9 --- /dev/null +++ b/readme.md @@ -0,0 +1,117 @@ +![ShowAndTell](https://github.com/litleCarl/ShowAndTell/blob/master/DemoImages/showAndTell.png) + +

+ + + +

+ +# Show and Tell: A Neural Image Caption Generator + +## Brief + +***Pull requests and issues:*** +@litleCarl + +A CoreML implementation of the image-to-text model described in the paper: + +"Show and Tell: Lessons learned from the 2015 MSCOCO Image Captioning +Challenge." + +Oriol Vinyals, Alexander Toshev, Samy Bengio, Dumitru Erhan. + +*IEEE transactions on pattern analysis and machine intelligence (2016).* + +Full text available at: http://arxiv.org/abs/1609.06647 + +## Demo + + + +## Usage + +### Simple use +```Swift +let showAndTell = ShowAndTell() +let results = showAndTell.predict(image: uiimage2predict, beamSize: 3, maxWordNumber: 30) +``` + + + +```Swift +// Parameter explaination +// image: The image to be used to generate the caption. +// beamSize: Max caption count in result to be reserved in beam search.(Affect the performance greatly) +// maxWordNumber: Max number of words in a sentence to be predicted. +class ShowAndTell { + ... + func predict(image: UIImage, beamSize: Int = 3, maxWordNumber:Int = 20) -> PriorityQueue + ... +} +``` + +## Benchmark (Tested on iPhone 7+, welcome pr for more decices) + + +
maxWordNumber = 20 maxWordNumber = 30
+ +beamSize | Time (ms) +---- | --- +1 | 480.12 +2 | 845.78 +3 | 1443.82 +4 | 2001.30 +5 | 2648.48 +6 | 3158.53 +7 | 4179.14 +8 | 4861.66 +9 | 6003.65 +10 | 7087.97 +11 | 8134.95 +12 | 9627.79 + + + +beamSize | Time (ms) +---- | --- +1 | 451.12 +2 | 1194.65 +3 | 1965.27 +4 | 2971.92 +5 | 3798.28 +6 | 4391.35 +7 | 5714.87 +8 | 6937.60 +9 | 8482.03 +10 | 10421.52 +11 | 12460.80 +12 | 13777.67 +
+ +### Line chart for Time vs Beam Size (When `maxWordNumber = 30`) + + +## Original Model +This coreml model is exported from keras which is trained with MSCOCO dataset for about 40k epoches. And presently it is not in the state of art yet. You may not use this in production. +I trained the dataset with only one GTX Force 1080Ti for about 48 hours and currently don't have more time to train on it.Hope for community to keep on it. + +## Requirements +- iOS 11.0+ +- Xcode 9.0+ (Swift 4.x) + +## TODO +- Train on the dataset to 100k epoches. (currently 40k) +- Open source origin model based on Keras which is trained with. + +## Thanks for third party lib in demo +- [GSMessages](https://github.com/wxxsw/GSMessages) +- [Swift-PriorityQueue](https://github.com/Bouke/Swift-PriorityQueue/) + +## Contact +- 曹佳鑫 (tsao)An iOS developer with experience in deep learning living in Shanghai. +- Pull requests and issues are welcome. +- Mail: cjx5813@foxmail.com + +## License + +ShowAndTell is available under the MIT license. See the LICENSE file for more info.